bzoj 1031: [JSOI2007]字符加密Cipher【后缀数组】

算是SA的裸题了
把串复制一遍接在原串后面,然后求SA,然后按着SA的顺序输出尾字符即可

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=200005;
int n,sa[N],wa[N],wb[N],wv[N],wsu[N];
char s[N];
bool cmp(int r[],int a,int b,int l)
{
	return r[a]==r[b]&&r[a+l]==r[b+l];
}
void saa(char r[],int n,int m)
{
	int *x=wa,*y=wb;
	for(int i=0;i<=m;i++)
		wsu[i]=0;
	for(int i=1;i<=n;i++)
		wsu[x[i]=r[i]]++;
	for(int i=1;i<=m;i++)
		wsu[i]+=wsu[i-1];
	for(int i=n;i>=1;i--)
		sa[wsu[x[i]]--]=i;
	for(int j=1,p=1;j<=n&&p<n;j<<=1,m=p)
	{
		p=0;
		for(int i=n-j+1;i<=n;i++)
			y[++p]=i;
		for(int i=1;i<=n;i++)
			if(sa[i]>j)
				y[++p]=sa[i]-j;
		for(int i=1;i<=n;i++)
			wv[i]=x[y[i]];
		for(int i=0;i<=m;i++)
			wsu[i]=0;
		for(int i=1;i<=n;i++)
			wsu[wv[i]]++;
		for(int i=1;i<=m;i++)
			wsu[i]+=wsu[i-1];
		for(int i=n;i>=1;i--)
			sa[wsu[wv[i]]--]=y[i];
		swap(x,y);
		x[sa[1]]=1;
		p=1;
		for(int i=2;i<=n;i++)
			x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p:++p;
	}
}
int main()
{
	scanf("%s",s+1);
	n=strlen(s+1);
	for(int i=1;i<=n;i++)
		s[i+n]=s[i];
	n<<=1;
	saa(s,n,200);
	for(int i=1;i<=n;i++)
		if(sa[i]<=n/2)
			printf("%c",s[sa[i]+(n>>1)-1]);
	return 0;
}
posted @ 2018-09-24 15:42  lokiii  阅读(118)  评论(0编辑  收藏  举报