UOJ #35. 后缀排序 后缀数组 模板

http://uoj.ac/problem/35

模板题，重新理了一遍关系。看注释吧。充分理解了倍增的意义，翻倍之后对上一次排序的利用是通过一种类似于队列的方式完成的。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=200010;
const int pl=50;
int sa[maxn+pl]={};//排名第i的是从sa[i]开始的数组
int rk[maxn+pl]={};//i的排名
int height[maxn+pl]={};//排名第i的与排名第i-1的最长相同前缀长度
int temp[maxn+pl]={};//暂时的排名
int cnt[maxn+pl]={};//第i种（字典序）前缀的有多少个（的前缀和）
int p[maxn+pl]={};//此次需要排列的sa的储存，处理了后缀长度不同的情况。
char ch[maxn+pl]={};
int siz;
bool equ(int x,int y,int l){ return rk[x]==rk[y]&&rk[x+l]==rk[y+l]; }
void SA(){
    for(int i=1;i<=siz;i++){rk[i]=ch[i];sa[i]=i;}
    for(int i,sig=255,l=0,pos=0;pos<siz;sig=pos){
        pos=0;
        for(i=siz-l+1;i<=siz;i++)p[++pos]=i;//根据上一次的sa决定p的储存顺序，没有排序则排在最前。
        for(i=1;i<=siz;i++) if(sa[i]>l) p[++pos]=sa[i]-l;//sa的计算是向字符串前端（左侧）拓展的
        for(i=1;i<=sig;i++) cnt[i]=0;
        for(i=1;i<=siz;i++) cnt[rk[p[i]]]++;
        for(i=1;i<=sig;i++) cnt[i]+=cnt[i-1];
        for(i=siz;i>0;i--) sa[cnt[rk[p[i]]]--]=p[i];//方便记忆板子的提示，只有这里是倒着扫的。
        pos=0;
        for(i=1;i<=siz;i++){
            if(equ(sa[i],sa[i-1],l))temp[sa[i]]=pos;
            else temp[sa[i]]=++pos;
        }
        for(int i=1;i<=siz;i++)rk[i]=temp[i];
        if(!l)l=1;
        else l<<=1;
    }int j=0;
    for(int i=1;i<=siz;i++){
        if(rk[i]==1){j=0;continue;}
        if(j)j--;
        while(ch[i+j]==ch[sa[rk[i]-1]+j]){
            j++;
        }height[rk[i]]=j;
    }
}
int main(){
    //freopen("a.in","r",stdin);
    scanf("%s",ch+1);siz=strlen(ch+1);
    SA();
    for(int i=1;i<=siz;i++)printf("%d ",sa[i]);
    cout<<endl;
    for(int i=2;i<=siz;i++)printf("%d ",height[i]);
    return 0;
}