[JSOI2007]字符加密 后缀数组
Code:
#include <cstdio>
#include <algorithm>
#include <cstring>
#define setIO(s) freopen(s".in","r",stdin)
#define maxn 1000000
using namespace std;
char str[maxn];
int n,m,rk[maxn],sa[maxn],tp[maxn],c[maxn],arr[maxn];
void qsort(){
for(int i=0;i<=m;++i) c[i]=0;
for(int i=1;i<=n;++i) ++c[rk[tp[i]]];
for(int i=1;i<=m;++i) c[i]+=c[i-1];
for(int i=n;i>=1;--i) sa[c[rk[tp[i]]]--]=tp[i];
}
void suffix(){
for(int i=1;i<=n;++i) rk[i]=arr[i],tp[i]=i;
qsort();
for(int k=1;k<=n;k<<=1){
int num=0;
for(int i=n-k+1;i<=n;++i) tp[++num]=i;
for(int i=1;i<=n;++i) if(sa[i]>k)tp[++num]=sa[i]-k;
qsort();
swap(rk,tp);
rk[sa[1]]=1;
num=1;
for(int i=2;i<=n;++i)
rk[sa[i]]=(tp[sa[i]]==tp[sa[i-1]]&&tp[sa[i-1]+k]==tp[sa[i]+k]) ? num : ++num;
if(num==n) break;
m=num;
}
}
int main(){
//setIO("input");
int cnt;
scanf("%s",str);
cnt=n=strlen(str);
m=200;
for(int i=n;i<=2*n-1;++i) str[i]=str[i-n];
n*=2;
for(int i=1;i<=n;++i) arr[i]=str[i-1];
suffix();
//printf("%d\n",cnt);
for(int i=1;i<=n;++i){
if(sa[i]>cnt) continue;
// printf("%d ",sa[i]+cnt-2);
putchar(str[sa[i]+cnt-2]);
}
return 0;
}