【tyvj1860】后缀数组

描述

我们定义一个字符串的后缀suffix(i)表示从s[i]到s[length(s)]这段子串。
后缀数组（Suffix array）SA[i]中存放着一个排列，满足suffix(sa[i])<suffix(sa[i+1]) 按照字典序方式比较
定义height[i]表示suffix(sa[i])与suffix(sa[i-1])之间的最长公共前缀长度，其中height[1]=0
你的任务就是求出SA和height这两个数组。字符串长度<=200000

输入格式

一行，为描述中的字符串（仅会出现小写字母）

输出格式

共两行，每行n个数，第一行为sa[i]，第二行为height[i]，其中每行的数均用空格隔开

测试样例1

输入

aabaaaab

输出

4 5 6 1 7 2 8 3 
0 3 2 3 1 2 0 1

题目告诉我们这是一个模板题，这的确就是一个模板题，geth函数还不是太理解。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<set>
#include<ctime>
#include<vector>
#include<cmath>
#include<algorithm>
#include<map>
#define inf 2000000000
#define N 200005
int n;
char ch[N];
int a[N],h[N];
int v[N];
int sa[2][N],rk[2][N];
int p,q,k;
void init(){
    scanf("%s",ch+1);
    n=strlen(ch+1);
    for (int i=1;i<=n;i++) a[i]=ch[i]-'a'+1;
}

void change(int sa[N],int rk[N],int SA[N],int RK[N]){
    for (int i=1;i<=n;i++) v[rk[sa[i]]]=i;//该排名最后出现的位置
    for (int i=n;i>=1;i--) if (sa[i]>k)
        SA[v[rk[sa[i]-k]]--]=sa[i]-k;//sa[i]-k代表一个字符串
    for (int i=n-k+1;i<=n;i++) SA[v[rk[i]]--]=i;//?
    for (int i=1;i<=n;i++)  RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i-1]]!=rk[SA[i]]||rk[SA[i-1]+k]!=rk[SA[i]+k]);
    
}

void work(){
     q=1,p=0;
    for (int i=1;i<=n;i++) v[a[i]]++;
    for (int i=1;i<=26;i++)v[i]+=v[i-1];
    for (int i=1;i<=n;i++) sa[p][v[a[i]]--]=i;
    for (int i=1;i<=n;i++) rk[p][sa[p][i]]=(rk[p][sa[p][i-1]])+(a[sa[p][i-1]]!=a[sa[p][i]]);
    k=1;
    while (k<n){
        change(sa[p],rk[p],sa[q],rk[q]);
        p^=1;q^=1;k=k<<1;
    }
    for (int i=1;i<=n;i++) printf("%d ",sa[p][i]);
}

void geth(){
    int k=0;
    for (int i=1;i<=n;i++)
    if (rk[p][i]==1)h[rk[p][i]]=0;
    else{
        int j=sa[p][rk[p][i]-1];
        while (a[i+k]==a[j+k])k++;
        h[rk[p][i]]=k;if (k>0)k--;
    }
}

int main(){
    init();
    work();
    geth();
    puts("");
    for(int i=1;i<=n;i++)printf("%d ",h[i]);
}