题目链接
bzoj 1031: [JSOI2007]字符加密Cipher
后缀数组裸题。
把字符串接到自己后面后直接排序。
输出结尾字符就好了QAQ
/**************************************************************
Problem: 1031
Language: C++
Result: Accepted
Time:1588 ms
Memory:21328 kb
****************************************************************/
#include<cstdio>
#include<cstring>
#include<algorithm>
const int maxn = 1000007;
int saf[maxn],sa[maxn],rank[maxn],cnt[maxn],s[maxn];
char a[maxn];
int n,m,st;
inline void rsort() {
for(int i=0;i<=m;++i) cnt[i]=0;
for(int i=1;i<=n;++i) cnt[rank[saf[i]]]++;
for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
for(int i=n;i>=1;--i) sa[cnt[rank[saf[i]]]--]=saf[i];
}
inline bool cmp(int *f,int x,int y,int w) {
return f[x]==f[y]&&f[x+w]==f[y+w];
}
void get_sa() {
for(int i=1;i<=n;++i) rank[i]=s[i],saf[i]=i;m=125;
rsort();
for(int p=0,w=1;p<n;w<<=1,m=p) {
p=0;
for(int i=n-w+1;i<=n;++i) saf[++p]=i;
for(int i=1;i<=n;++i)
if(sa[i]>w) saf[++p]=sa[i]-w;
rsort();
std::swap(rank,saf);rank[sa[1]]=p=1;
for(int i=2;i<=n;++i) rank[sa[i]]=cmp(saf,sa[i],sa[i-1],w)? p:++p;
}
for(int i=1;i<=n;++i) if(sa[i]<=st)printf("%c",a[sa[i]+st-1]);
}
int main() {
scanf("%s",a+1);
st=n=strlen(a+1);
for(int i=1;i<n;++i)a[i+n]=a[i];
n=n*2-1;
for(int i=1;i<=n;++i) s[i]=a[i],s[i+n]=s[i];
s[0]=s[n+1]=-1;
get_sa();
return 0;
}