【BZOJ1031】字符加密 后缀数组

后缀数组模板题

用的罗学长惊为天人的模版。。。当时搅了整整一周才明白。。。

/**************************************************************
    Problem: 1031
    User: RicardoWang
    Language: C++
    Result: Accepted
    Time:820 ms
    Memory:7912 kb
****************************************************************/
 
#include<cstdlib>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
#define maxn 200005
char s[2*maxn];
int len,n,m;
void Init()
{
     scanf("%s",s+1);
     for(len=1;;len++)
     {
        if(s[len]=='\0'||s[len]=='\n')
        {
            len--; break;
        }
     }
     n=len;
     for(int i=1;i<=n;i++)s[i+n]=s[i];
     n=n*2;
     m=500;
     return ;
}
int wa[maxn*2],wb[maxn*2],sa[maxn*2],cc[maxn*2];
void makesa()
{
    int *y=wa,*x=wb,*t;
    int cnt;
    for(int i=1;i<=n;i++)x[i]=s[i];
    for(int i=1;i<=n;i++)cc[x[i]]++;
    for(int i=1;i<=m;i++)cc[i]=cc[i]+cc[i-1];
    for(int i=n;i>=1;i--)sa[cc[x[i]]--]=i;
    for(int k=1;k<=n;k=k*2)
    {
        cnt=0;
        for(int i=n-k+1;i<=n;i++)y[++cnt]=i;
        for(int i=1;i<=n;i++)if(sa[i]>k)y[++cnt]=sa[i]-k;
        for(int i=1;i<=m;i++)cc[i]=0;
        for(int i=1;i<=n;i++)cc[x[y[i]]]++;
        for(int i=1;i<=m;i++)cc[i]=cc[i]+cc[i-1];
        for(int i=n;i>=1;i--)sa[cc[x[y[i]]]--]=y[i];
        t=x; x=y; y=t;x[sa[1]]=1;cnt=1;
        for(int i=2;i<=n;i++)
        {
            x[sa[i]]=(y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k])? cnt:++cnt;
        }
        if(cnt==n)break; m=cnt;
    }
    return ;
}
int main()
{
    //freopen("in.txt","r",stdin);
    Init();
    makesa();
    for(int i=1;i<=n;i++)if(sa[i]<=n/2)
    {
        putchar(s[sa[i]+n/2-1]);
    }
    putchar('\n');
    return 0;
}