forked from t3nsor/codebook
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Z-algorithm.cpp
39 lines (35 loc) · 951 Bytes
/
Z-algorithm.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
void getZarr(string str, int Z[]) {
int n = str.length();
int L, R, k;
L = R = 0;
for (int i = 1; i < n; ++i) {
if (i > R) {
L = R = i;
while (R < n && str[R - L] == str[R])
R++;
Z[i] = R - L;
R--;
} else {
k = i - L;
if (Z[k] < R - i + 1)
Z[i] = Z[k];
else {
L = i;
while (R < n && str[R - L] == str[R])
R++;
Z[i] = R - L;
R--;
}
}
}
}
void search(string text, string pattern) {
string concat = pattern + "$" + text;
int l = concat.length();
int Z[l];
getZarr(concat, Z);
for (int i = 0; i < l; ++i) {
if (Z[i] == pattern.length())
cout << "Pattern found at index " << i - pattern.length() - 1 << endl;
}
}