洛谷P3809 【模板】后缀排序

题目背景

这是一道模板题。

题目描述

读入一个长度为 nn 的由大小写英文字母或数字组成的字符串，请把这个字符串的所有非空后缀按字典序从小到大排序，然后按顺序输出后缀的第一个字符在原串中的位置。位置编号为 11 到 nn 。

输入输出格式

输入格式：

 

一行一个长度为 nn 的仅包含大小写英文字母或数字的字符串。

 

输出格式：

 

一行，共n个整数，表示答案。

 

输入输出样例

输入样例#1： 复制

ababa

输出样例#1： 复制

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();    
}