Tyvj_P1860 后缀数组(后缀数组模板题)

Tyvj传送门

时间: 1000ms / 空间: 131072KiB / Java类名: Main
描述
我们定义一个字符串的后缀suffix(i)表示从s[i]到s[length(s)]这段子串。
后缀数组(Suffix array)SA[i]中存放着一个排列,满足suffix(sa[i])< suffix(sa[i+1]) 按照字典序方式比较
定义height[i]表示suffix(sa[i])与suffix(sa[i-1])之间的最长公共前缀长度,其中height[1]=0
你的任务就是求出SA和height这两个数组。字符串长度<=200000

输入格式
一行,为描述中的字符串(仅会出现小写字母)

输出格式
共两行,每行n个数,第一行为sa[i],第二行为height[i],其中每行的数均用空格隔开

测试样例1
输入

aabaaaab

输出

4 5 6 1 7 2 8 3
0 3 2 3 1 2 0 1

Sol:
发现了自己程序的一个Bug….2333QuQ

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 200005
int t1[N],t2[N],c[N];
int sa[N],rk[N],ht[N];
int n,m=27,len;int s[N];
void sort(int *x,int *y){
    for(int i=0;i<m;i++) c[i]=0;
    for(int i=0;i<n;i++) c[x[y[i]]]++;
    for(int i=1;i<m;i++) c[i]+=c[i-1];
    for(int i=n-1;~i;--i) sa[--c[x[y[i]]]]=y[i];
}
void DA(){
    int *x=t1,*y=t2;
    for(int i=0;i<m;i++) c[i]=0;
    for(int i=0;i<n;i++) c[x[i]=s[i]]++;
    for(int i=1;i<m;i++) c[i]+=c[i-1];
    for(int i=n-1;~i;i--) sa[--c[x[i]]]=i;
    for(int k=1,p=0;k<=n;k<<=1,p=0){
        for(int i=n-k;i<n;i++) y[p++]=i;
        for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        sort(x,y);swap(x,y);m=1,x[sa[0]]=0;
        for(int i=1;i<n;i++) x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&
            ((sa[i]+k>=n?-1:y[sa[i]+k])==(sa[i-1]+k>=n?-1:y[sa[i-1]+k]))?m-1:m++;
        if(m>n) break;
    }
    for(int i=0;i<n;i++) rk[sa[i]]=i;
    for(int i=0,j,k=0;i<n;ht[rk[i++]]=k){
        if(!rk[i]) continue;
        j=sa[rk[i]-1],k=k?k-1:0;
        while(i+k<n&&j+k<n&&s[i+k]==s[j+k]) k++;
    }ht[0]=0;
}
char str[64];
void print(int x){
    int l=0;if(!x){putchar('0'),putchar(' ');return;}
    while(x) str[++l]=x%10+'0',x/=10;while(l) putchar(str[l--]);
    putchar(' ');
}
int main(){
    char ch=getchar();while(ch>'z'||ch<'a') ch=getchar();
    while(ch>='a'&&ch<='z') s[n++]=ch-'a',ch=getchar();
    DA();
    for(int i=0;i<n;i++) print(sa[i]+1);putchar('\n');
    for(int i=0;i<n;i++) print(ht[i]);
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值