[BZOJ1031][JSOI2007]字符加密Cipher

原题地址

后缀数组模板题.

build函数内为后缀数组必背模板(只含sa数组和rank数组，不包含h数组和height数组).
15个for…

AC code:

#include <cstdio>
#include <cstring>
const int K=1<<10;
const int N=200010;
int  l;
int  c[N],rk[N],sa[N],r1[N],r2[N],t[N];
char s[N];

void build(){
    for(int i=0;i<l;i++) c[s[i]]=1;
    for(int i=1;i<K;i++) c[i]+=c[i-1];
    for(int i=0;i<l;i++) rk[i]=c[s[i]];
    for(int i=1,j;rk[sa[l]]!=l;i<<=1){
        for(j=0;j<l;j++) r1[j]=j+i<l?rk[j+i]:0;
        for(j=0;j<l;j++) c[j]=0;
        for(j=0;j<l;j++) c[r1[j]]++;
        for(j=1;j<l;j++) c[j]+=c[j-1];
        for(j=0;j<l;j++) t[c[r1[j]]--]=j;
        for(j=0;j<l;j++) c[j]=0;
        for(j=0;j<l;j++) c[rk[j]]++;
        for(j=1;j<l;j++) c[j]+=c[j-1];
        for(j=l;j>0;j--) sa[c[rk[t[j]]]--]=t[j];
        for(r2[sa[1]]=1,j=2;j<=l;j++) r2[sa[j]]=r2[sa[j-1]]+(rk[sa[j]]!=rk[sa[j-1]]||r1[sa[j]]!=r1[sa[j-1]]);
        for(j=0;j<l;j++) rk[j]=r2[j];
    }
}

int main(){
    scanf("%s",s);
    l=strlen(s);
    for(int i=l;i<(l<<1);i++) s[i]=s[i-l];
    l<<=1;
    build();
    for(int i=1;i<=l;i++){
        if(sa[i]<(l>>1)) printf("%c",s[sa[i]+(l>>1)-1]);
    }
    printf("\n");

    return 0;
}