【uoj35】后缀排序

这是一道模板题。

读入一个长度为 n 的由小写英文字母组成的字符串，请把这个字符串的所有非空后缀按字典序从小到大排序，然后按顺序输出后缀的第一个字符在原串中的位置。位置编号为 1 到 n。

除此之外为了进一步证明你确实有给后缀排序的超能力，请另外输出 n−1 个整数分别表示排序后相邻后缀的最长公共前缀的长度。

输入格式
一行一个长度为 n 的仅包含小写英文字母的字符串。

输出格式
第一行 n 个整数，第 i 个整数表示排名为 i 的后缀的第一个字符在原串中的位置。

第二行 n−1 个整数，第 i 个整数表示排名为 i 和排名为 i+1 的后缀的最长公共前缀的长度。

样例一
input
C++

ababa
1
ababa
output
C++

5 3 1 4 2
1 3 0 2
1
2
5 3 1 4 2
1 3 0 2
explanation
排序后结果为：

a
aba
ababa
ba
baba
限制与约定
1≤n≤10^5

题解
后缀数组，对于求hight，有一个小定理，hight[i]>=hight[i-1]-1 所以记录前一个的答案判断即可。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#define N 500005
#define inf 1000000000
using namespace std;
int a[N],b[N],rk[N],sa[N],tsa[N],sta[200005],stb[200005],s[100005],hight[100005];
char ch[100005];
int n;
inline int read()
{
    int x=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
void solve_sa()
{
    for (int i=1;i<=n;i++)
    {
        s[i]=ch[i]-'a'+1;
        sta[s[i]]++;
    }
    for (int i=1;i<=26;i++)
        sta[i]+=sta[i-1];
    for (int i=n;i>=1;i--) sa[sta[s[i]]--]=i;
    rk[sa[1]]=1;
    for (int i=2;i<=n;i++) rk[sa[i]]=rk[sa[i-1]]+(s[sa[i-1]]!=s[sa[i]]);
    for (int L=1;rk[sa[n]]!=n;L<<=1)
    {
        for (int i=0;i<=n;i++) sta[i]=stb[i]=0;
        for (int i=1;i<=n;i++) a[i]=rk[i],b[i]=rk[i+L];
        for (int i=1;i<=n;i++) sta[a[i]]++,stb[b[i]]++;
        for (int i=1;i<=n;i++) sta[i]+=sta[i-1],stb[i]+=stb[i-1];
        for (int i=n;i>=1;i--) tsa[stb[b[i]]--]=i;
        for (int i=n;i>=1;i--) sa[sta[a[tsa[i]]]--]=tsa[i];
        rk[sa[1]]=1;
        for (int i=2;i<=n;i++) rk[sa[i]]=rk[sa[i-1]]+(a[sa[i-1]]!=a[sa[i]]||b[sa[i-1]]!=b[sa[i]]);
    }
    for (int i=1;i<=n;i++) printf("%d ",sa[i]);
}
void solve_h()
{
    int j,k=0;
    for (int i=1;i<=n;hight[rk[i++]]=k)
    {
        for (k?k--:0,j=sa[rk[i]-1];s[i+k]==s[j+k];k++);
    }   
    for (int i=2;i<=n;i++) printf("%d ",hight[i]);
}
int main()
{
    scanf("%s",ch+1);
    n=strlen(ch+1);
    solve_sa();
    puts("");
    solve_h();
    return 0;
}