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