TyvjP1860 后缀数组

来源:互联网 发布:mac休眠后继续下载吗 编辑:程序博客网 时间:2024/06/08 11:56

描述
我们定义一个字符串的后缀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

分析:
练模板,目前最优美的模板

ps.Tyvj的页面真是超级美,不知道比luogu,bzoj高到哪里去了
这里写图片描述

这里写代码片#include<cstdio>#include<cstring>#include<iostream>using namespace std;const int N=200010;int sa[N],a[N],b[N],cc[N];int len,hei[N],rak[N];char s[N];int cmp(int *y,int a,int b,int k){    int ra1=y[a];    int rb1=y[b];    int ra2=a+k>=len ? -1:y[a+k];    int rb2=b+k>=len ? -1:y[b+k];    return ra1==rb1&&ra2==rb2;}void make_sa(){    int i,j,m,p;    int *x=a,*y=b,*t;    m=26;    for (i=0;i<m;i++) cc[i]=0;    for (i=0;i<len;i++) cc[x[i]=s[i]-'a']++;    for (i=1;i<m;i++) cc[i]+=cc[i-1];    for (i=len-1;i>=0;i--) sa[--cc[x[i]]]=i;    for (int k=1;k<=len;k<<=1)    {        p=0;        for (i=len-k;i<len;i++) y[p++]=i;  //        for (i=0;i<len;i++) if (sa[i]>=k) y[p++]=sa[i]-k;        for (i=0;i<m;i++) cc[i]=0;        for (i=0;i<len;i++) ++cc[x[y[i]]];        for (i=1;i<m;i++) cc[i]+=cc[i-1];        for (i=len-1;i>=0;i--) sa[--cc[x[y[i]]]]=y[i];        t=x;x=y;y=t;        x[sa[0]]=0;        p=1;  //        for (i=1;i<len;i++)            x[sa[i]]=cmp(y,sa[i],sa[i-1],k) ? p-1:p++;        if (p>=len) break;        m=p;    } }void make_hei(){    int i,k=0;    for (i=0;i<len;i++) rak[sa[i]]=i;    hei[0]=0;    for (i=0;i<len;i++)    {        if (!rak[i]) continue;        int j=sa[rak[i]-1];        if (k)           k--;        while (s[i+k]==s[j+k]&&i+k<len&&j+k<len) k++;          //i+k<len&&j+k<len        hei[rak[i]]=k;    }}int main(){    scanf("%s",&s);    len=strlen(s);    make_sa();    make_hei();    for (int i=0;i<len;i++) printf("%d ",sa[i]+1);    puts("");    for (int i=0;i<len;i++) printf("%d ",hei[i]);    return 0;}
原创粉丝点击