-
Notifications
You must be signed in to change notification settings - Fork 5
Expand file tree
/
Copy pathstring_suffix_array.cpp
More file actions
57 lines (54 loc) 路 1.75 KB
/
Copy pathstring_suffix_array.cpp
File metadata and controls
57 lines (54 loc) 路 1.75 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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
template <typename T> vector<int> suffix_array(const T &S) {
int N = int(S.size());
vector<int> suffix(N), classes(N);
for (int i = 0; i < N; i++) {
suffix[i] = N - 1 - i;
classes[i] = S[i];
}
stable_sort(suffix.begin(), suffix.end(),
[&S](int i, int j) { return S[i] < S[j]; });
for (int len = 1; len < N; len *= 2) {
vector<int> c(classes);
for (int i = 0; i < N; i++) {
bool same = i && suffix[i - 1] + len < N &&
c[suffix[i]] == c[suffix[i - 1]] &&
c[suffix[i] + len / 2] == c[suffix[i - 1] + len / 2];
classes[suffix[i]] = same ? classes[suffix[i - 1]] : i;
}
vector<int> cnt(N), s(suffix);
for (int i = 0; i < N; i++) {
cnt[i] = i;
}
for (int i = 0; i < N; i++) {
int s1 = s[i] - len;
if (s1 >= 0)
suffix[cnt[classes[s1]]++] = s1;
}
}
return suffix;
}
template <typename T> vector<int> lcp_array(const vector<int> &sa, const T &S) {
int N = int(S.size());
vector<int> rank(N), lcp(N - 1);
for (int i = 0; i < N; i++)
rank[sa[i]] = i;
int pre = 0;
for (int i = 0; i < N; i++) {
if (rank[i] < N - 1) {
int j = sa[rank[i] + 1];
while (max(i, j) + pre < int(S.size()) && S[i + pre] == S[j + pre])
++pre;
lcp[rank[i]] = pre;
if (pre > 0)
--pre;
}
}
return lcp;
}
/// Complexity: O(|N|*log(|N|))
// Usage:
// Index: 012345
// string some_string = "banana";
// vector<int> suffix = suffix_array(some_string)
// suffix{5, 3, 1, 0, 4, 2}
// 5:a, 3:ana, 1:anana, 0:banana, 4:na, 2:nana