【BZOJ-1031】字符加密Cipher 后缀数组

#include<cstdio> 
#include<cstring> 
using namespace std; 
#define maxn 2000010 
int wa[maxn],wb[maxn],wv[maxn],ws[maxn]; 
char S[maxn]; int SA[maxn]; 
int cmp(int *r,int a,int b,int l)   
{ 
    return r[a]==r[b]&&r[a+l]==r[b+l]; 
} 
void DA(char *r,int *sa,int n,int m)   
{   
    int p,*x=wa,*y=wb,*t;   
    for(int i=0; i<m; i++) ws[i]=0;   
    for(int i=0; i<n; i++) ws[x[i]=r[i]]++; 
    for(int i=1; i<m; i++) ws[i]+=ws[i-1];   
    for(int i=n-1; i>=0; i--) sa[--ws[x[i]]]=i; 
    p=1;for(int j=1;p<n;j*=2,m=p)   
    {   
        p=0; for(int i=n-j;i<n;i++) y[p++]=i; 
        for(int i=0; i<n; i++) if(sa[i]>=j) y[p++]=sa[i]-j;   
        for(int i=0; i<n; i++) wv[i]=x[y[i]];   
        for(int i=0; i<m; i++) ws[i]=0;   
        for(int i=0; i<n; i++) ws[wv[i]]++;   
        for(int i=1; i<m; i++) ws[i]+=ws[i-1];   
        for(int i=n-1; i>=0; i--) sa[--ws[wv[i]]]=y[i]; 
        t=x,x=y,y=t,p=1,x[sa[0]]=0; 
        for(int i=1; i<n;i++)   
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;   
    }    
}   
int main() 
{ 
    scanf("%s",S); 
    int len=strlen(S); 
    for (int i=len; i<len*2; i++) S[i]=S[i-len]; 
    len=len*2+1; S[len]=‘\n‘; 
    DA(S,SA,len,300); 
    for (int i=0; i<len; i++) 
        if (SA[i]<(len>>1)) 
            putchar(S[SA[i]+(len>>1)-1]);  
    return 0; 
}