【tyvj1860】后缀数组

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<cstdlib>
 5 using namespace std;
 6 const int N=200050;
 7 int Sa[N],wu[N],wv[N],wa[N],wb[N],R[N],Rank[N],Height[N],rank[N];
 8 char s[N];
 9 int cmp(int *r,int a,int b,int l){return r[a]==r[b] && r[a+l]==r[b+l];}
10 inline void SA(int *r,int *sa,int n,int m){
11     int *x=wa,*y=wb;
12     for(int i=0;i<m;++i)wu[i]=0;
13     for(int i=0;i<n;++i)++wu[x[i]=r[i]];
14     for(int i=1;i<m;++i)wu[i]+=wu[i-1];
15     for(int i=n-1;i>=0;--i)sa[--wu[x[i]]]=i;
16     for(int j=1,p=0;p<n;j<<=1,m=p){
17         p=0;
18         for(int i=n-j;i<n;++i)y[p++]=i;
19         for(int i=0;i<n;++i)if(sa[i]>=j)y[p++]=sa[i]-j;
20         for(int i=0;i<m;++i)wu[i]=0;
21         for(int i=0;i<n;++i)++wu[wv[i]=x[y[i]]];
22         for(int i=0;i<m;++i)wu[i]+=wu[i-1];
23         for(int i=n-1;i>=0;--i)sa[--wu[wv[i]]]=y[i];
24         swap(x,y);
25         p=1;
26         x[sa[0]]=0;
27         for(int i=1;i<n;++i)x[sa[i]]=((y[sa[i]]==y[sa[i-1]]) && y[sa[i]+j]==y[sa[i-1]+j]) ? p-1 : p++;
28     }
29 }
30 inline void Get_Height(int *r,int n){
31     for(int i=1;i<=n;++i)rank[Sa[i]]=i;
32     for(int i=0,j=0,k=0;i<n;Height[rank[i++]]=k)
33         for(k?--k:0,j=Sa[rank[i]-1];r[i+k]==r[j+k];++k);
34 }
35 inline void output(int x){
36     if(!x)return;output(x/10);
37     putchar((x%10)+‘0‘);
38 }
39 int main(){
40     scanf("%s",s);
41     int len=strlen(s);
42     for(int i=0;i<len;++i)R[i]=s[i]-‘a‘+1;
43     R[len]=0;
44     SA(R,Sa,len+1,30);
45     Get_Height(R,len);
46     for(int i=1;i<=len;++i)output(Sa[i]+1),putchar(‘ ‘);puts("");
47     for(int i=1;i<=len;++i){
48         if(!Height[i])putchar(‘0‘);
49         else output(Height[i]);putchar(‘ ‘);
50     }puts("");
51     return 0;
52 }