[学习笔记]后缀数组 [JSOI2007] bzoj1031 字符加密

……其实后缀数组学很久了今天敲的时候发现好久不写都不会写了就翻blog发现没有写过……所以来用个模板题屯一波板子……
这个题显然就是展开后缀数组即可。
代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#define N 200010
using namespace std;
int wa[N],wb[N],cnt[N],v[N],x[N],y[N];
char s[N];int sa[N],h[N],rk[N],n;
inline int cmp(int *a,int x,int y,int l)
{
	return a[x]==a[y]&&a[x+l]==a[y+l];
}
inline int getSA(char *s,int n,int m)
{
	int i,j,p,*x=wa,*y=wb;
	memset(wa,0,sizeof(wa));
	memset(wb,0,sizeof(wb));
	for(i=1;i<=m;i++) cnt[i]=0;
	for(i=1;i<=n;i++) cnt[x[i]=s[i]]++;
	for(i=1;i<=m;i++) cnt[i]+=cnt[i-1];
	for(i=n;i;i--) sa[cnt[x[i]]--]=i;
	for(j=1,p=0;p<=n;j<<=1,m=p)
	{
		for(p=0,i=n-j+1;i<=n;i++) y[++p]=i;
		for(i=1;i<=n;i++)
			if(sa[i]>j) y[++p]=sa[i]-j;
		for(i=1;i<=m;i++) cnt[i]=0;
		for(i=1;i<=n;i++) cnt[v[i]=x[y[i]]]++;
		for(i=1;i<=m;i++) cnt[i]+=cnt[i-1];
		for(i=n;i;i--) sa[cnt[v[i]]--]=y[i];
		for(swap(x,y),p=i=1;i<=n;i++)
			x[sa[i]]=(cmp(y,sa[i],sa[i-1],j)?p-1:(p++));
	}
	for(i=1;i<=n;i++) rk[sa[i]]=i;
	for(i=1,p=0;i<=n;h[rk[i++]]=p)
		for((p?p--:0),j=sa[rk[i]-1];s[i+p]==s[j+p];p++);
	return 0;
}
int main()
{
	scanf("%s",s+1),n=strlen(s+1);
	for(int i=1;i<n;i++) s[n+i]=s[i];
	getSA(s,n*2-1,1010);
	for(int i=1;i<=2*n-1;i++)
		if(sa[i]<=n) printf("%c",s[sa[i]+n-1]);
	printf("\n");return 0;
}