后缀数组练习题bzoj 1031 后缀数组模板题目

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Kamisama123/article/details/76218298

题目戳这里http://www.lydsy.com/JudgeOnline/problem.php?id=1031
这是一道显然的后缀数组模板题目。
这道题我们发现只要求出这几个循环的字符串排列就好了,那么我们显然想到把它扩展2倍,然后求一边rank和SA数组,这样我们得到了需要的前一半rank数组,把它离散一下就是这个循环圈的排列顺序了。
AC代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define N 200005
using namespace std;
int n,m,i,p,w,tp[N],a[N],rank[N],SA[N],tax[N],ans[N],aa[N];
char s[N];
void Rsort(){
    for(i=0;i<=m;i++)tax[i]=0;
    for(i=1;i<=n;i++)tax[rank[tp[i]]]++;
    for(i=1;i<=m;i++)tax[i]+=tax[i-1];
    for(i=n;i>=1;--i)SA[tax[rank[tp[i]]]--]=tp[i];
}
bool cmp(int *f,int x,int y,int w){return f[x]==f[y]&&f[x+w]==f[y+w];}
void sa(){
    for(i=1;i<=n;i++)rank[i]=a[i],tp[i]=i;
    m=127;Rsort();
    for(w=1,p=1;p<n;m=p,w*=2){
        for(p=0,i=n-w+1;i<=n;i++)tp[++p]=i;
        for(i=1;i<=n;i++)if(SA[i]>w)tp[++p]=SA[i]-w;
        Rsort();swap(tp,rank);rank[SA[1]]=p=1;
        for(i=2;i<=n;i++)rank[SA[i]]=cmp(tp,SA[i],SA[i-1],w)?p:++p;
    }
}
void init(){
    scanf("%s",s);
    n=strlen(s);
    for(i=0;i<n;i++)
    a[i+1]=s[i];
    for(i=n+1;i<=2*n;i++)
    a[i]=a[i-n];
    n*=2;
}
int main(){
    init();
    sa();
    n/=2;
    for(i=1;i<=n;i++)
    ans[i]=rank[i];
    sort(ans+1,ans+n+1);
    int x,y;
    for(i=1;i<=n;i++){
        x=lower_bound(ans+1,ans+n+1,rank[i])-ans;
        if(i==1)y=n;else y=i-1;
        aa[x]=y;
    }
    for(i=1;i<=n;i++)
    printf("%c",s[aa[i]-1]);
}

处理答案输出有点复杂了,表在意细节。

展开阅读全文

没有更多推荐了,返回首页