洛谷P3809 【模板】尾碼排序

題目背景

這是一道模板題。

題目描述

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

輸入輸出格式

輸入格式：

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

輸出格式：

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

輸入輸出樣例

ababa

5 3 1 4 2

說明

n <= 10^6n<=106

更新了一下自己寫的尾碼數組

看了一晚上還是有些地方不理解。

明天繼續搞

// luogu-judger-enable-o2
#include<cstring>
#include<iostream>
#include<cstdio>
using namespace std;
const int MAXN=1e6+10;
int sa[MAXN],tax[MAXN],rak[MAXN],tp[MAXN],a[MAXN],N,M;
char s[MAXN];
int pd(int *f,int x,int y,int l)
{
    return (f[x]==f[y]&&f[x+l]==f[y+l]);
}
void Q_sort()
{
    for(int i=0;i<=M;i++) tax[i]=0;
    for(int i=1;i<=N;i++) tax[rak[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];
}
void S_sort()
{
    M=127;
    for(int i=1;i<=N;i++) rak[i]=a[i],tp[i]=i;Q_sort();
    for(int w=1,p=1;p<N;w=w+w,M=p)
    {        
        p=0;
        for(int i=1;i<=w;i++) tp[++p]=N-w+i;
        for(int i=1;i<=N;i++)
            if(sa[i]>w) tp[++p]=sa[i]-w;
        Q_sort();swap(tp,rak);rak[sa[1]]=p=1;    
        for(int i=2;i<=N;i++) 
            rak[sa[i]]=pd(tp,sa[i],sa[i-1],w)?p:++p;
    }
    for(int i=1;i<=N;i++) 
        printf("%d ",sa[i]);
}
int main()
{
    #ifdef WIN32
    freopen("a.in","r",stdin);
    #else
    #endif
    scanf("%s",s);
    N=strlen(s);
    for(int i=0;i<N;i++) a[i+1]=s[i]-48;
    S_sort();    
}