TyvjP1860 后缀数组

描述

我们定义一个字符串的后缀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

分析：
练模板，目前最优美的模板

ps.Tyvj的页面真是超级美，不知道比luogu，bzoj高到哪里去了

这里写代码片
#include<cstdio>
#include<cstring>
#include<iostream>

using namespace std;

const int N=200010;
int sa[N],a[N],b[N],cc[N];
int len,hei[N],rak[N];
char s[N];

int cmp(int *y,int a,int b,int k)
{
    int ra1=y[a];
    int rb1=y[b];
    int ra2=a+k>=len ? -1:y[a+k];
    int rb2=b+k>=len ? -1:y[b+k];
    return ra1==rb1&&ra2==rb2;
}

void make_sa()
{
    int i,j,m,p;
    int *x=a,*y=b,*t;
    m=26;
    for (i=0;i<m;i++) cc[i]=0;
    for (i=0;i<len;i++) cc[x[i]=s[i]-'a']++;
    for (i=1;i<m;i++) cc[i]+=cc[i-1];
    for (i=len-1;i>=0;i--) sa[--cc[x[i]]]=i;
    for (int k=1;k<=len;k<<=1)
    {
        p=0;
        for (i=len-k;i<len;i++) y[p++]=i;  //
        for (i=0;i<len;i++) if (sa[i]>=k) y[p++]=sa[i]-k;
        for (i=0;i<m;i++) cc[i]=0;
        for (i=0;i<len;i++) ++cc[x[y[i]]];
        for (i=1;i<m;i++) cc[i]+=cc[i-1];
        for (i=len-1;i>=0;i--) sa[--cc[x[y[i]]]]=y[i];
        t=x;x=y;y=t;
        x[sa[0]]=0;
        p=1;  //
        for (i=1;i<len;i++)
            x[sa[i]]=cmp(y,sa[i],sa[i-1],k) ? p-1:p++;
        if (p>=len) break;
        m=p;
    } 
}

void make_hei()
{
    int i,k=0;
    for (i=0;i<len;i++) rak[sa[i]]=i;
    hei[0]=0;
    for (i=0;i<len;i++)
    {
        if (!rak[i]) continue;
        int j=sa[rak[i]-1];
        if (k)
           k--;
        while (s[i+k]==s[j+k]&&i+k<len&&j+k<len) k++;  
        //i+k<len&&j+k<len
        hei[rak[i]]=k;
    }
}

int main()
{
    scanf("%s",&s);
    len=strlen(s);
    make_sa();
    make_hei();
    for (int i=0;i<len;i++) printf("%d ",sa[i]+1);
    puts("");
    for (int i=0;i<len;i++) printf("%d ",hei[i]);
    return 0;
}