【后缀数组】uoj#35. 后缀排序

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define N 100001
int sa
,t
,t2
,tong
,n;
char s
;
inline bool cmp(int *y,int i,int k)
{
return (y[sa[i-1]]==y[sa[i]])&&((sa[i-1]+k>=n?-1:y[sa[i-1]+k])==(sa[i]+k>=n?-1:y[sa[i]+k]));
}
//构造字符串s的后缀数组，每个字符值必须为0~m-1，字符串下标为0~n-1
void build_sa(int range)
{
int *x=t,*y=t2;
//基数排序
memset(tong,0,sizeof(int)*range);//清空桶
for(int i=0;i<n;++i) tong[x[i]=s[i]]++;//把s拷贝到x中，之后插入桶
for(int i=1;i<range;++i) tong[i]+=tong[i-1];//将桶处理成前缀和
for(int i=n-1;i>=0;--i) sa[--tong[x[i]]]=i;
for(int k=1;k<=n;k<<=1)
{
int p=0;
//直接利用sa数组排序第二关键字
for(int i=n-k;i<n;++i) y[p++]=i;
for(int i=0;i<n;++i) if(sa[i]>=k) y[p++]=sa[i]-k;
//基数排序第一关键字
memset(tong,0,sizeof(int)*range);
for(int i=0;i<n;++i) tong[x[y[i]]]++;
for(int i=0;i<range;++i) tong[i]+=tong[i-1];
for(int i=n-1;i>=0;--i) sa[--tong[x[y[i]]]]=y[i];
//根据sa和y数组计算新的x数组
swap(x,y);
p=1; x[sa[0]]=0;
for(int i=1;i<n;++i) x[sa[i]]= cmp(y,i,k) ? p-1 : p++;
if(p>=n) break;
range=p;
}
}
int rank
,lcp
;
void get_lcp()
{
int k=0;
for(int i=0;i<n;++i) rank[sa[i]]=i;
for(int i=0;i<n;++i) if(rank[i])
{
if(k) --k;
int j=sa[rank[i]-1];
while(s[i+k]==s[j+k]) ++k;
lcp[rank[i]]=k;
}
}
int main()
{
scanf("%s",s);
n=strlen(s);
build_sa('z'+1);
for(int i=0;i<n;++i) printf("%d ",sa[i]+1);
puts("");
if(n>1) get_lcp();
for(int i=1;i<n;++i) printf("%d ",lcp[i]);
return 0;
}