P3809 【模版】字尾排序

題目背景

這是一道模版題。

題目描述

讀入一個長度為 nn 的由大小寫英文字母或數字組成的字串，請把這個字串的所有非空字尾按字典序從小到大排序，然後按順序輸出字尾的第一個字元在原串中的位置。位置編號為 11 到 nn。

輸入輸出格式

輸入格式：

 

一行一個長度為 nn 的僅包含大小寫英文字母或數字的字串。

 

輸出格式：

 

一行，共n個整數，表示答案。

 

輸入輸出樣例

輸入樣例#1：

ababa

輸出樣例#1：

5 3 1 4 2

說明

n <= 10^6n<=10​6​​

 

看了一下午的字尾陣列

基數排序的思想我懂啊。，

倍增的思想我懂啊。。

字尾陣列的思想我也懂啊。。

為毛程式碼一行都看不懂啊。。。。。。。。。。。。。。。。。。。。。。。

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int MAXN=1000001;
void read(int &n)
{
    char c='+';int x=0;bool flag=0;
    while(c<'0'||c>'9'){c=getchar();if(c=='-')flag=1;}
    while(c>='0'&&c<='9'){x=x*10+(c-48);c=getchar();}
    flag==1?n=-x:n=x;
}
int tax[MAXN];// 基數排序的輔助陣列
int tp[MAXN];//基數排序的第二關鍵字
int a[MAXN];// 字串陣列
int sa[MAXN];//
int rak[MAXN]; 
int n,m;
void qsort()
{
    for(int i=0;i<=m;i++)    tax[i]=0;
    for(int i=1;i<=n;i++)    tax[rak[tp[i]]]++;
    for(int i=1;i<=m;i++)    tax[i]+=tax[i-1];
    for(int i=n;i>=1;i--)    sa[tax[rak[tp[i]]]--]=tp[i];
}
int comp(int *f,int x,int y,int l)
{
    return (f[x]==f[y]&&f[x+l]==f[y+l]);
}
void suffix_ar()
{
    for(int i=1;i<=n;i++)
        rak[i]=a[i],tp[i]=i;
    m=127,qsort();
    for(int w=1,p=1,i;p<n;w+=w,m=p)
    {
        for(p=0,i=n-w+1;i<=n;i++)
            tp[++p]=i;
        for(int i=1;i<=n;i++)
            if(sa[i]>w)
                tp[++p]=sa[i]-w;
        qsort(),swap(rak,tp),rak[sa[1]]=p=1;
        for(int i=2;i<=n;i++)
            rak[sa[i]]=comp(tp,sa[i],sa[i-1],w)? p:++p;
    }
    for(int i=1;i<=n;i++)
        printf("%d ",sa[i]);
}
int main()
{
    string s;
    cin>>s;
    n=s.length();
    for(int i=0;i<(s.length());i++)
        a[i+1]=s[i]-48;
    suffix_ar();
    return 0;
}