menu zcmimi's blog
LG 3809 【模板】后缀排序
查看原题

点击跳转

#include<bits/stdc++.h>
#define fur(i,x,y) for(int i=x;i<=y;++i)
const int N=1000001;
char s[N];
int n,m,sa[N],rnk[N],tp[N],tax[N];
void qsort(){
    fur(i,0,m)tax[i]=0;
    fur(i,1,n)++tax[rnk[i]];
    fur(i,1,m)tax[i]+=tax[i-1];
    for(int i=n;i;--i)
        sa[tax[rnk[tp[i]]]--]=tp[i];
}
void suffix(){
    m=75;
    fur(i,1,n)rnk[i]=s[i]-'0'+1,tp[i]=i;
    qsort();
    for(int w=1,p=0;p<n;m=p,w<<=1){
        p=0;
        fur(i,1,w)tp[++p]=n-w+i;
        fur(i,1,n)if(sa[i]>w)
            tp[++p]=sa[i]-w;
        qsort();
        std::swap(tp,rnk);
        rnk[sa[1]]=p=1;
        fur(i,2,n)
            rnk[sa[i]]=p+=tp[sa[i-1]]^tp[sa[i]]||tp[sa[i-1]+w]^tp[sa[i]+w];
    }
}
/* -------------------height--------------------- */
int height[N];
void get_height(){
    for(int i=1,k=0;i<=n;++i){
        if(k)--k;
        int j=sa[rnk[i]-1];
        while(s[i+k]==s[j+k]&&i+k<=n&&j+k<=n)++k;
        height[rnk[i]]=k; 
    }
}
/* ---------------------------------------------- */
int main(){
    scanf("%s",s+1),n=strlen(s+1);
    suffix();
    fur(i,1,n)printf("%d ",sa[i]);
}
LG 3809 【模板】后缀排序

评论加载中...

Search
search