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