Update sa.cc
This commit is contained in:
parent
d0330f151b
commit
cf752da653
40
string/sa.cc
40
string/sa.cc
|
@ -1,39 +1,28 @@
|
||||||
#include <algorithm>
|
#include "../include.hh"
|
||||||
#include <cstdio>
|
|
||||||
#include <cstring>
|
|
||||||
#include <iostream>
|
|
||||||
|
|
||||||
using namespace std;
|
constexpr int N = 1e6 + 10;
|
||||||
|
|
||||||
const int N = 1000010;
|
|
||||||
|
|
||||||
char s[N];
|
char s[N];
|
||||||
// key1[i] = rk[id[i]](作为基数排序的第一关键字数组)
|
|
||||||
int n, sa[N], rk[N], oldrk[N << 1], id[N], key1[N], cnt[N], height[N];
|
int n, sa[N], rk[N], oldrk[N << 1], id[N], key1[N], cnt[N], height[N];
|
||||||
|
|
||||||
bool cmp(int x, int y, int w) {
|
bool cmp(int x, int y, int w) {
|
||||||
return oldrk[x] == oldrk[y] && oldrk[x + w] == oldrk[y + w];
|
return oldrk[x] == oldrk[y] && oldrk[x + w] == oldrk[y + w];
|
||||||
}
|
}
|
||||||
|
|
||||||
int main() {
|
void calc_sa() {
|
||||||
int i, m = 127, p, w;
|
|
||||||
|
|
||||||
scanf("%s", s + 1);
|
|
||||||
n = strlen(s + 1);
|
n = strlen(s + 1);
|
||||||
|
int i, m = 127, p, w;
|
||||||
// calc sa[]
|
|
||||||
for (i = 1; i <= n; ++i) ++cnt[rk[i] = s[i]];
|
for (i = 1; i <= n; ++i) ++cnt[rk[i] = s[i]];
|
||||||
for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
|
for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
|
||||||
for (i = n; i >= 1; --i) sa[cnt[rk[i]]--] = i;
|
for (i = n; i >= 1; --i) sa[cnt[rk[i]]--] = i;
|
||||||
|
|
||||||
for (w = 1;; w <<= 1, m = p) { // m=p 就是优化计数排序值域
|
for (w = 1;; w <<= 1, m = p) {
|
||||||
for (p = 0, i = n; i > n - w; --i) id[++p] = i;
|
for (p = 0, i = n; i > n - w; --i) id[++p] = i;
|
||||||
for (i = 1; i <= n; ++i)
|
for (i = 1; i <= n; ++i)
|
||||||
if (sa[i] > w) id[++p] = sa[i] - w;
|
if (sa[i] > w) id[++p] = sa[i] - w;
|
||||||
|
|
||||||
memset(cnt, 0, sizeof(cnt));
|
memset(cnt, 0, sizeof(cnt));
|
||||||
for (i = 1; i <= n; ++i) ++cnt[key1[i] = rk[id[i]]];
|
for (i = 1; i <= n; ++i) ++cnt[key1[i] = rk[id[i]]];
|
||||||
// 注意这里px[i] != i,因为rk没有更新,是上一轮的排名数组
|
|
||||||
|
|
||||||
for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
|
for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
|
||||||
for (i = n; i >= 1; --i) sa[cnt[key1[i]]--] = id[i];
|
for (i = n; i >= 1; --i) sa[cnt[key1[i]]--] = id[i];
|
||||||
|
@ -44,16 +33,11 @@ int main() {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
// calc height[]
|
|
||||||
for (i = 1, k = 0; i <= n; ++i) {
|
int main() {
|
||||||
if (rk[i] == 0) continue;
|
untie;
|
||||||
if (k) --k;
|
cin >> (s + 1); // array s starts from index 1
|
||||||
while (s[i + k] == s[sa[rk[i] - 1] + k]) ++k;
|
calc_sa();
|
||||||
height[rk[i]] = k;
|
for (int i = 1; i <= n; ++i) cout << sa[i] << " \n"[i == n];
|
||||||
}
|
|
||||||
|
|
||||||
for (i = 1; i <= n; ++i) printf("%d ", sa[i]);
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue