tyvj1860 後綴數組 模板

注意常數優化（優化輸出等）。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN=2e5+10;
int n,sa[MAXN],tmp[MAXN],rank[MAXN],sec[MAXN],cnt[MAXN],height[MAXN];
char s[MAXN];
void get_sa()
{
    for(int i=1;i<=n;i++) rank[i]=s[i]-'a'+1;
    for(int i=1;i<=n;i++) cnt[rank[i]]++;
    for(int i=1;i<=26;i++)cnt[i]+=cnt[i-1];
    for(int i=1;i<=n;i++) sa[cnt[rank[i]]--]=i;

    for(int k=1;k<=n;k<<=1)
    {
        int top=0;
        for(int i=n-k+1;i<=n;i++) sec[++top]=i;
        for(int i=1;i<=n;i++) if(sa[i]>k) sec[++top]=sa[i]-k;

        for(int i=0;i<=n;i++) cnt[i]=0;
        for(int i=1;i<=n;i++) cnt[rank[i]]++;
        for(int i=1;i<=n;i++) cnt[i]+=cnt[i-1];
        for(int i=n;i>=1;i--) sa[cnt[rank[sec[i]]]--]=sec[i];

        top=tmp[sa[1]]=1;
        for(int i=2;i<=n;i++)
        {
            if(rank[sa[i]]!=rank[sa[i-1]]||rank[sa[i]+k]!=rank[sa[i-1]+k]) top++;
            tmp[sa[i]]=top;
        }
        for(int i=1;i<=n;i++) rank[i]=tmp[i];
    }
    int k=0;
    for(int i=1;i<=n;i++)
    {
        if(k) k--;
        while(i+k<=n&&sa[rank[i]-1]+k<=n&&s[i+k]==s[sa[rank[i]-1]+k]) k++;
        height[rank[i]]=k;
    }
}
void putt(int x)
{
    if(!x)return;
    char c=x%10+'0';
    putt(x/10);
    putchar(c);
}
int main()
{
    scanf("%s",s+1);
    n=strlen(s+1);
    get_sa();
    for(int i=1;i<=n;i++)
    {
        putt(sa[i]);
        putchar(' ');
    }
    puts("");
    for(int i=1;i<=n;i++)
    {
        if(!height[i]) putchar('0');
        else putt(height[i]);
        putchar(' ');
    }
    return 0;
}