您的位置:首页 > 其它

【模板】后缀数组

2017-10-16 16:15 369 查看
见这里

#include<map>
#include<cmath>
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
#define N 200010
int n,v
,sa1
,rk1
,sa2
,rk2
,ht
;
char s
;
int *sa,*rk,*SA,*RK;
int main()
{
scanf("%d",&n);
scanf("%s",s+1);
rk=rk1;RK=rk2;sa=sa1;SA=sa2;
int i,k,j;
for(i=1;i<=n;i++) v[s[i]]++;
for(i=1;i<=200;i++) v[i]+=v[i-1];
for(i=n;i;i--) sa[v[s[i]]--]=i;
for(i=1;i<=n;i++) rk[sa[i]]=rk[sa[i-1]]+(s[sa[i]]!=s[sa[i-1]]);
for(k=1;k<=n;k<<=1)
{
for(i=1;i<=n;i++) v[rk[sa[i]]]=i;
for(i=n;i;i--) if(sa[i]>k) SA[v[rk[sa[i]-k]]--]=sa[i]-k;
for(i=n-k+1;i<=n;i++) SA[v[rk[i]]--]=i;
for(i=1;i<=n;i++) RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i]]!=rk[SA[i-1]]||rk[SA[i]+k]!=rk[SA[i-1]+k]);
swap(sa,SA);swap(rk,RK);
}
k=0;
for(i=1;i<n;i++)
{
j=rk[i]-1;
while(s[sa[rk[i]]+k]==s[sa[j]+k]) k++;
ht[rk[i]]=k;
if(k) k--;
}
for(i=1;i<=n;i++) printf("%d ",sa[i]-1);
puts("");
for(i=1;i<=n;i++) printf("%d ",rk[i]);
puts("");
for(i=2;i<=n;i++) printf("%d ",ht[i]);
puts("");
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: