【后缀数组】BZOJ1031(JSOI2007)[字符加密Cipher]题解

题目描述

给出一个字符串（设长度为len），造出len个长度为len的字符串，造字符串方法如下：

造出的几个字符串为：JSOI07,SOI07J,OI07JS,I07JSO,07JSOI,7JSOI0
将造出的字符串排序，输出排序过后每个字符串的最后一个字符。

解题报告

这道题明显就是字符串快速排序，但是如何快速排序？我们比较熟悉的字符串快速排序算法有后缀数组，这道题可不可以用后缀数组实现呢？其实是可以的！
以这个字符串为例：
123
我们知道后缀数组只能排后缀，如果直接这么排肯定是错的，因为这道题需要环状，那么我们就接环：
123123
排序过后得到：
123
123123
23
23123
3
3123
但是，长度<=6（等价于该后缀编号>6，也就是接环多出来的节点）的我们不要，因为那些是多余的，筛选过后剩下：
123123
23123
3123
那么答案就是这几个字符串的第三个字符，即312

示例程序

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxl=200000;

int len,SA[maxl+5],rk[maxl+5],t[maxl+5],ha[maxl+5];
char s[maxl+5];

bool Eoln(char ch) {return ch==10||ch==13||ch==EOF;}
int reads(char *s) //char[]读入优化
{
    int len=0;char ch=getchar();if (ch==EOF) return EOF;
    s[++len]=ch;while (!Eoln(s[len])) s[++len]=getchar();s[len--]=0;
    return len;
}
void make_SA(char *s) //构造SA
{
    int len=strlen(s+1),MAX=0;
    memset(ha,0,sizeof(ha));
    for (int i=1;i<=len;i++) ha[rk[i]=s[i]]++,MAX=max(MAX,rk[i]);
    for (int i=1;i<=MAX;i++) ha[i]+=ha[i-1];
    for (int i=len;i>=1;i--) SA[ha[rk[i]]--]=i;
    for (int k=1;k<=len;k<<=1)
    {
        int p=0;
        for (int i=len-k+1;i<=len;i++) t[++p]=i;
        for (int i=1;i<=len;i++) if (SA[i]>k) t[++p]=SA[i]-k;
        memset(ha,0,sizeof(ha));
        for (int i=1;i<=len;i++) ha[rk[t[i]]]++;
        for (int i=1;i<=MAX;i++) ha[i]+=ha[i-1];
        for (int i=len;i>=1;i--) SA[ha[rk[t[i]]]--]=t[i];
        memcpy(t,rk,sizeof(t));
        p=0;rk[SA[1]]=++p;
        for (int i=2;i<=len;i++)
            if (t[SA[i-1]]==t[SA[i]]&&t[SA[i-1]+k]==t[SA[i]+k]) rk[SA[i]]=p; else rk[SA[i]]=++p;
        if (p==len) break;
        MAX=p;
    }
}
int main()
{
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    len=reads(s);for (int i=1;i<=len;i++) s[len+i]=s[i];s[(len*=2)+1]=0; //读入并接环
    make_SA(s);
    for (int i=1;i<=len;i++) if (SA[i]<=len/2) putchar(*(s+SA[i]+len/2-1));
    //输出SA[i]<=len/2（长度>len/2）的第len/2个字符
    putchar('\n');
    return 0;
}