您的位置:首页 > 其它

后缀数组 nbut1171 后缀数组

2016-02-05 15:06 190 查看
传送门:点击打开链接

题意:完美的后缀数组模板测试题

思路:学习后缀数组可以看集训队的一篇论文,我觉得写的非常棒点击打开链接

#include<map>
#include<set>
#include<cmath>
#include<ctime>
#include<stack>
#include<queue>
#include<cstdio>
#include<cctype>
#include<string>
#include<vector>
#include<cstring>
#include<iomanip>
#include<iostream>
#include<algorithm>
#include<functional>
#define fuck(x) cout<<"["<<x<<"]"
#define FIN freopen("input.txt","r",stdin)
#define FOUT freopen("output.txt","w+",stdout)
using namespace std;
typedef long long LL;
typedef pair<int, int>PII;

const int MX = 1e4 + 5;
const int mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;

/*
复杂度O(nlogn)
字符串r要从0开始
n自动+1无需再管,返回的SA,R,H的下标都是0~n
其中多包括了一个空字符串
SA后缀数组,R名次数组,H高度数组
*/
char s[MX];
int SA[MX], R[MX], H[MX];
int wa[MX], wb[MX], wv[MX], wc[MX];
bool cmp(int *r, int a, int b, int l) {
return r[a] == r[b] && r[a + l] == r[b + l];
}
void Suffix(char *r, int m = 128) {
int n = strlen(r) + 1;
int i, j, p, *x = wa, *y = wb, *t;
for(i = 0; i < m; i++) wc[i] = 0;
for(i = 0; i < n; i++) wc[x[i] = r[i]]++;
for(i = 1; i < m; i++) wc[i] += wc[i - 1];
for(i = n - 1; i >= 0; i--) SA[--wc[x[i]]] = i;
for(j = 1, p = 1; p < n; j *= 2, m = p) {
for(p = 0, i = n - j; i < n; i++) y[p++] = i;
for(i = 0; i < n; i++) if(SA[i] >= j) y[p++] = SA[i] - j;
for(i = 0; i < n; i++) wv[i] = x[y[i]];
for(i = 0; i < m; i++) wc[i] = 0;
for(i = 0; i < n; i++) wc[wv[i]]++;
for(i = 1; i < m; i++) wc[i] += wc[i - 1];
for(i = n - 1; i >= 0; i--) SA[--wc[wv[i]]] = y[i];
for(t = x, x = y, y = t, p = 1, x[SA[0]] = 0, i = 1; i < n; i++) {
x[SA[i]] = cmp(y, SA[i - 1], SA[i], j) ? p - 1 : p++;
}
}
int k = 0; n--;
for(i = 0; i <= n; i++) R[SA[i]] = i;
for(i = 0; i < n; i++) {
if(k) k--;
j = SA[R[i] - 1];
while(r[i + k] == r[j + k]) k++;
H[R[i]] = k;
}
}

int main() {
//FIN;
bool first = true;
while(~scanf("%s", s)) {
Suffix(s);
int n = strlen(s);

if(first) first = false;
else printf("\n");

for(int i = 1; i <= n; i++) {
int p = SA[i];
printf("%d %s\n", p + 1, s + p);
}
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: