-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDC3.cpp
More file actions
43 lines (38 loc) · 1.31 KB
/
DC3.cpp
File metadata and controls
43 lines (38 loc) · 1.31 KB
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
40
41
42
43
char str [MAX_N];
int N, m, SA [MAX_N], LCP [MAX_N];
int x [MAX_N], y [MAX_N], w [MAX_N], c [MAX_N];
inline bool cmp (const int a, const int b, const int l) { return (y [a] == y [b] && y [a + l] == y [b + l]); }
void Sort () {
for (int i = 0; i < m; ++i) w [i] = 0;
for (int i = 0; i < N; ++i) ++w [x [y [i]]];
for (int i = 0; i < m - 1; ++i) w [i + 1] += w [i];
for (int i = N - 1; i >= 0; --i) SA [--w [x [y [i]]]] = y [i];
}
void DA () {
++N;
for (int i = 0; i < N; ++i) x [i] = str [i], y[i] = i;
Sort ();
for (int i, j = 1, p = 1; p < N; j <<= 1, m = p) {
for (p = 0, i = N - j; i < N; i++) y [p++] = i;
for (int k = 0; k < N; ++k) if (SA [k] >= j) y [p++] = SA [k] - j;
Sort ();
for (swap (x, y), p = 1, x [SA [0]] = 0, i = 1; i < N; ++i) x [SA [i]] = cmp (SA [i - 1], SA [i], j) ? p - 1 : p++;
}
for (int i = 1; i < N; ++i) SA [i - 1] = SA [i]; --N;
}
void kasaiLCP () {
for (int i = 0; i < N; ++i) c [SA [i]] = i;
LCP [0] = 0;
for (int i = 0, h = 0; i < N; ++i) if (c[i] > 0) {
int j = SA [c [i] - 1];
while (i + h < N && j + h < N && str [i + h] == str [j + h]) ++h;
LCP [c [i]] = h;
if (h > 0) --h;
}
}
void suffixArray () {
m = 256;
N = strlen (str);
DA ();
kasaiLCP ();
}