【bzoj 1031】字符加密Cipher（後綴數組模板）

傳送門~

解題思路

後綴數組模板。
dalao的詳細講解：鏈接
代碼：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<cmath>
#include<cstdlib>
using namespace std;
int sa[200005];
int rank[200005];
int x[200005];
int y[200005];
int tmp[200005];
int sum[200005];
char s[200005];
int len,tot;
void get_rank(){
    for(int i=1;i<=len;i++) ++sum[s[i]];
    for(int i=1;i<=127;i++) sum[i]+=sum[i-1];
    for(int i=len;i>=1;i--) tmp[sum[s[i]]--]=i;
    int now=0;
    for(int i=1;i<=len;i++){
        if(i==1 || s[tmp[i]]!=s[tmp[i-1]]) ++now;
        rank[tmp[i]]=now;
    }
}
void ji_sort(int *key,int *order){
    for(int i=0;i<=len;i++) sum[i]=0;
    for(int i=1;i<=len;i++) ++sum[key[i]];
    for(int i=1;i<=len;i++) sum[i]+=sum[i-1];
    for(int i=len;i>=1;i--) tmp[sum[key[order[i]]]--]=order[i];
    for(int i=1;i<=len;i++) order[i]=tmp[i];
}
void get_hou(){
    get_rank();
    for(int i=1;i<=len;i<<=1){
        for(int j=1;j<=len;j++){
            x[j]=rank[j];
            y[j]=j+i>len ? 0 : rank[j+i];
            sa[j]=j;
        }
        ji_sort(y,sa);
        ji_sort(x,sa);
        int now=0;
        for(int j=1;j<=len;j++){
            if(j==1 || x[sa[j]]!=x[sa[j-1]] || y[sa[j]]!=y[sa[j-1]]) now++;
            rank[sa[j]]=now;
        }
    }
}
int main(){
    int ylen;
    scanf("%s",s+1);
    ylen=len=strlen(s+1);
    for(int i=1;i<=len;i++) s[len+i]=s[i];
    s[len+len+1]=s[1];
    len=len*2+1;get_hou();
    for(int i=1;i<=len;i++) 
    if(sa[i]<=ylen) printf("%c",s[sa[i]+ylen-1]);
    return 0;
}