【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]);
}