P3809 【模板】后缀排序

【模板】后缀排序 - 洛谷

题目背景

这是一道模板题。

题目描述

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

输入格式

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

输出格式

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

输出格式

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

输入输出样例

输入 #1

ababa

输出 #1

5 3 1 4 2

#include<bits/stdc++.h>
using namespace std;
int len;
const int N=1e6+5;
char str[N];
int sa[N];
int ranks[N];
int temp[N];
int k;
bool cmp(int a,int b)
{
    if(ranks[a]!=ranks[b])return ranks[a]<ranks[b];
    int ra=a+k<=len?ranks[a+k]:-1;
    int rb=b+k<=len?ranks[b+k]:-1;
    return ra<rb;
}
void solve()
{
    scanf("%s",str);
    len=strlen(str);
    for(int i=0;i<=len;i++)
    {
        sa[i]=i;
        ranks[i]=i<len?str[i]:-1;
    }
    for( k=1;k<=len;k<<=1)
    {
        sort(sa,sa+len+1,cmp);
        temp[sa[0]]=0;
        for(int i=1;i<=len;i++)
        {
            temp[sa[i]]=temp[sa[i-1]]+(cmp(sa[i-1],sa[i])?1:0);
        }
        for(int i=0;i<=len;i++)ranks[i]=temp[i];
    }
    for(int i=1;i<=len;i++)printf("%d ",sa[i]+1);
}
int main()
{
    solve();
    return 0;
}