1. 程式人生 > >洛谷P3809 【模板】後綴排序

洛谷P3809 【模板】後綴排序

esp oid 輸入格式 字符串 自己 後綴數組 pen log ios

題目背景

這是一道模板題。

題目描述

讀入一個長度為 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(); }

洛谷P3809 【模板】後綴排序