[省选前题目整理][BZOJ 1031][JSOI 2007]字符加密Cipher(后缀数组)

题目链接

http://www.lydsy.com/JudgeOnline/problem.php?id=1031

思路

根据题目要求，就是要给JSOI07这样给定的字符串的所有变换后的串进行排序，并按名次从低到高输出每个变换后的串的最后一位字符。那么首先把JSOI07这个字符串断环为链，变成JSOI07JSOI07，并删去最后一个字符，变成JSOI07JSOI0，防止排序后出现两个JSOI07的情况，然后对这个新的串做后缀数组，得到所有后缀的排序，实际上这个排序就是符合题目要求的排序。设原来题目给出的串长度为len，因为后缀的排序是字典序，而前len个后缀的前len位字符都是不同的(如果有相同的，那么两个前len位相同的后缀的相对顺序是无所谓的)，因此这个后缀的排序就是相当于按照每个后缀前len位的字典序的排序。
那么对于前len名后缀sa[x]=i,依次输出在新串中下标为i+len的字符即可。

代码

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>

#define MAXN 1100000

using namespace std;

char s[MAXN],ans[MAXN];
int num[MAXN];
int sa[MAXN],height[MAXN],rank[MAXN];
int wa[MAXN],wb[MAXN],wv[MAXN],cnt[MAXN];

bool cmp(int *r,int a,int b,int c)
{
    return (r[a]==r[b])&&(r[a+c]==r[b+c]);
}

void SA(int *r,int n,int m)
{
    int i,j,p;
    int *x=wa,*y=wb;
    for(i=0;i<m;i++) cnt[i]=0;
    for(i=0;i<n;i++) cnt[(x[i]=r[i])]++;
    for(i=1;i<m;i++) cnt[i]+=cnt[i-1];
    for(i=n-1;i>=0;i--) sa[--cnt[x[i]]]=i;
    for(j=1,p=1;p<n;j*=2,m=p)
    {
        for(p=0,i=n-j;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
        for(i=0;i<n;i++) wv[i]=x[y[i]];
        for(i=0;i<m;i++) cnt[i]=0;
        for(i=0;i<n;i++) cnt[wv[i]]++;
        for(i=1;i<m;i++) cnt[i]+=cnt[i-1];
        for(i=n-1;i>=0;i--) sa[--cnt[wv[i]]]=y[i]; //!!!!!
        swap(x,y);
        for(p=1,x[sa[0]]=0,i=1;i<n;i++)
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++; //!!!!!!
    }
}

void calc(int *r,int n)
{
    int i,j,k=0;
    for(i=0;i<n;i++) rank[sa[i]]=i;
    for(i=0;i<n;height[rank[i++]]=k)
        for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
}

int main()
{
    scanf("%s",s);
    int len=strlen(s);
    for(int i=0;i<len;i++)
        num[i]=s[i];
    for(int i=0;i<len;i++)
        num[i+len]=s[i];
    len*=2;
    num[len-1]=0;
    SA(num,len,255);
    //calc(num,len/2);
    for(int i=0;i<len;i++) rank[sa[i]]=i;
    for(int i=0;i<len/2;i++)
        ans[rank[i]]=(char)num[i+len/2-1];
    for(int i=0;i<len;i++)
        if(ans[i])
            printf("%c",ans[i]);
    return 0;
}