[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;
}