[UOJ35]後綴排序

描述
這是一道模板題。
讀入一個長度爲 nn 的由小寫英文字母組成的字符串，請把這個字符串的所有非空後綴按字典序從小到大排序，然後按順序輸出後綴的第一個字符在原串中的位置。位置編號爲 11 到 nn。
除此之外爲了進一步證明你確實有給後綴排序的超能力，請另外輸出 n−1n−1 個整數分別表示排序後相鄰後綴的最長公共前綴的長度。
輸入格式
一行一個長度爲 nn 的僅包含小寫英文字母的字符串。
輸出格式
第一行 nn 個整數，第 ii 個整數表示排名爲 ii 的後綴的第一個字符在原串中的位置。
第二行 n−1n−1 個整數，第 ii 個整數表示排名爲 ii 和排名爲 i+1i+1 的後綴的最長公共前綴的長度。
樣例一
input
ababa
output
5 3 1 4 2
1 3 0 2
explanation
排序後結果爲：
a
aba
ababa
ba
baba
限制與約定
1≤n≤1051≤n≤105
時間限制：1s1s
空間限制：256MB

Code
集SAM，ST，SA與一體

Solution

#include <bits/stdc++.h>
using namespace std;
#define right Right
const int MaxN=200100;
char s[MaxN];
int right[MaxN],len[MaxN],pos[MaxN],nxt[MaxN][26],fa[MaxN];
int up[MaxN],w[MaxN],ver[MaxN],ht[MaxN],rk[MaxN];
int sa[MaxN],sa_n=0,n;
struct Node{int v;Node *nxt;}pool[MaxN],*tail=pool,*g[MaxN];

inline void addedge(int u,int v){tail->v=v;tail->nxt=g[u];g[u]=tail++;}
struct sam{
    int root,last,cnt,n;
    sam(){root=last=++cnt;}
    void insert(int c,int r){
        int np=++cnt,p=last;last=np;
        right[np]=len[np]=len[p]+1; pos[np]=r;
        for(;p&&!nxt[p][c];nxt[p][c]=np,p=fa[p]);
        if (!p) fa[np]=root;
        else if (len[nxt[p][c]]==len[p]+1) fa[np]=nxt[p][c];
        else{
            int nq=++cnt,q=nxt[p][c];len[nq]=len[p]+1;right[nq]=right[q];
            memcpy(nxt[nq],nxt[q],sizeof nxt[q]);fa[nq]=fa[q];
            fa[np]=fa[q]=nq;
            for(;p&&nxt[p][c]==q;nxt[p][c]=nq,p=fa[p]);
        }
    }
    void build(){
        scanf("%s",s+1);n=strlen(s+1);
        for(int i=n;i;i--)insert(s[i]-'a',i);
    }
    void sort(){
        for(int i=1;i<=cnt;i++){
            up[i]=s[n+1-right[i]+len[fa[i]]];
            w[up[i]]++;
        }
        for(int i=1;i<='z';i++) w[i]+=w[i-1];
        for(int i=1;i<=cnt;i++) ver[w[up[i]]--]=i;
    }
    void build_ST(){for(int i=cnt;i>=2;i--)addedge(fa[ver[i]],ver[i]);}
}SAM;
void dfs(int x){
    if(pos[x])sa[++sa_n]=pos[x];
    for(Node *p=g[x];p;p=p->nxt)dfs(p->v);
}
int main(){
    SAM.build();SAM.sort();SAM.build_ST();dfs(1);

    n=sa_n; for(int i=1;i<=n;i++)rk[sa[i]]=i;
    for(int i=1,k=0;i<=n;i++){
        for(int j=sa[rk[i]-1];s[i+k]==s[j+k];k++);
        ht[rk[i]]=k?k--:k;
    }
    for(int i=1;i<=n;i++)printf("%d%c",sa[i]," \n"[i==n]);
    for(int i=2;i<=n;i++)printf("%d%c",ht[i]," \n"[i==n]);
    return 0;
}