【Tyvj3500】【BZOJ1031】字符加密，后缀数组

Time:2016.05.22

Author:xiaoyimi

转载注明出处谢谢

传送门1

传送门2

思路：

非常裸的题目了，连height数组都不用

把原串后面再接上一个原串，得到一个长度为原串两倍的新串，对它建立后缀数组即可，而且在原串中相对位置一样的两个后缀一定是挨在一起的，不会影响sa和rank数组，最后的答案就是排名从小到大，输出开头下标在原串长度内的答案就可以了

注意：无

代码：

#include<bits/stdc++.h>
#define M 200004
using namespace std;
char s[M];
int len,w[M],sa[M],rank[M],tmp[M],cnt[M],id[M];
void SA(int len,int up)
{
int p=0,d=1,*rk=rank,*t=tmp;
for (int i=0;i<up;i++) cnt[i]=0;
for (int i=0;i<len;i++) cnt[rk[i]=w[i]]++;
for (int i=1;i<up;i++) cnt[i]+=cnt[i-1];
for (int i=len-1;i>=0;i--) sa[--cnt[rk[i]]]=i;
for (;;)
{
for (int i=len-d;i<len;i++) id[p++]=i;
for (int i=0;i<len;i++)
if (sa[i]>=d) id[p++]=sa[i]-d;
for (int i=0;i<up;i++) cnt[i]=0;
for (int i=0;i<len;i++) cnt[t[i]=rk[id[i]]]++;
for (int i=1;i<up;i++) cnt[i]+=cnt[i-1];
for (int i=len-1;i>=0;i--) sa[--cnt[t[i]]]=id[i];
swap(rk,t);
p=1;
rk[sa[0]]=0;
for (int i=0;i<len-1;i++)
if (sa[i]+d<len&&sa[i+1]+d<len&&t[sa[i]+d]==t[sa[i+1]+d]&&t[sa[i]]==t[sa[i+1]])
rk[sa[i+1]]=p-1;
else rk[sa[i+1]]=p++;
if(p>=len) break;
up=p;p=0;d<<=1;
}
}
main()
{
scanf("%s",s);
len=strlen(s);
int maxn=0;
for (int i=0;i<len;i++)
w[i]=s[i],
maxn=max(maxn,w[i]);
for (int i=0;i<len;i++)
w[i+len]=s[i];
SA(len<<1,maxn+1);
for (int i=0;i<len<<1;i++)
if (sa[i]<len)
printf("%c",w[sa[i]+len-1]);
}