3172: [Tjoi2013]单词 后缀数组

来源:互联网 发布:在淘宝买东西怎么付钱 编辑:程序博客网 时间:2024/04/28 01:06

写了一发后缀数组,二分究竟哪里写错了QAQ,无奈改成了暴力过了。
后缀数组的作法比较显然,比起fail树来好想多了。。(我太弱啦)。
把各个串连起来之后从每个位置向两侧扩展,只要满足height[i]>=单词长度就可以。最后求出长度输出。
无奈之下把logN改成了N。。

#include<bits/stdc++.h>int pos[205],l[205];int n,len;char ch[1001005],s[1001005];int cc[1001005],rank[1001005],height[1001005],sa[1001005],t1[1001005],t2[1001005];int f[1001005][21];using namespace std;inline bool cmp(int *y,int a,int b,int k){    int arank1=y[a];    int brank1=y[b];    int arank2=a+k>=len?-1:y[a+k];    int brank2=b+k>=len?-1:y[b+k];    return arank1==brank1&&arank2==brank2;}inline void make_sa(){    int *x=t1,*y=t2;    int m=256;    for (int i=0;i<m;i++) cc[i]=0;    for (int i=0;i<len;i++) ++cc[x[i]=s[i]];    for (int i=1;i<m;i++) cc[i]+=cc[i-1];    for (int i=len-1;~i;i--) sa[--cc[x[i]]]=i;    for (int k=1;k<len;k<<=1)    {        int p=0;        for (int i=len-k;i<len;i++) y[p++]=i;        for (int i=0;i<len;i++)            if (sa[i]>=k) y[p++]=sa[i]-k;        for (int i=0;i<m;i++) cc[i]=0;        for (int i=0;i<len;i++) ++cc[x[y[i]]];        for (int i=1;i<m;i++) cc[i]+=cc[i-1];        for (int i=len-1;~i;i--) sa[--cc[x[y[i]]]]=y[i];        swap(x,y);        m=1; x[sa[0]]=0;        for (int i=1;i<len;i++)            x[sa[i]]=cmp(y,sa[i-1],sa[i],k)?m-1:m++;        if (m>=len) break;    }}inline void make_height(){    for (int i=0;i<len;i++) rank[sa[i]]=i;    int k=0; height[0]=0;    for (int i=0;i<len;i++)    {        if (!rank[i]) continue;        int j=sa[rank[i]-1];        if (k) k--;        while (s[j+k]==s[i+k]) k++;        height[rank[i]]=k;    }}inline void make_st(){    for (int i=0;i<len;i++) f[i][0]=height[i];    for (int j=1;j<=20;j++)        for (int i=0;i<len;i++)            if (i+(1<<j)-1<len)                 f[i][j]=min(f[i][j-1],f[i+(1<<(j-1))][j-1]);}inline int ask_min(int l,int r){    int k=log2(r-l+1);    return min(f[l][k],f[r-(1<<k)+1][k]);}   /*inline int findl(int pos,int length){    int l=1,r=rank[pos],ans=0;    while (l<=r)    {        int mid=l+r>>1;        if (ask_min(mid,rank[pos])>=length) ans=mid,r=mid-1; else l=mid+1;    }    if (ans) return rank[pos]-ans+1; return ans;}inline int findr(int pos,int length){    int l=rank[pos]+1,r=len-1,ans=0;    while (l<=r)    {        int mid=l+r>>1;        if (ask_min(rank[pos]+1,mid)>=length) ans=mid,l=mid+1; else r=mid-1;    }    if (ans) return ans-rank[pos]; return ans;}*/int main(){    scanf("%d",&n);    for (int i=1;i<=n;i++)    {        pos[i]=len;        scanf("%s",ch);        l[i]=strlen(ch);        for (int j=0;j<l[i];j++)            s[len++]=ch[j];        if (i!=n) s[len++]=32;    }    make_sa();    make_height();    make_st();//  for (int i=0;i<len;i++) printf("%s|||%d\n",s+sa[i],height[i]); //  for (int i=1;i<=n;i++)//  {   printf("%d %d \n ", findr(pos[i],l[i]), findl(pos[i],l[i]));    //  printf("%d\n",findr(pos[i],l[i])+findl(pos[i],l[i])+1);    for (int i=1;i<=n;i++)    {        int L,R;        for (L=rank[pos[i]];height[L]>=l[i];L--);        for (R=rank[pos[i]];height[R+1]>=l[i];R++);        printf("%d\n",R-L+1);    }    return 0;}
0 0
原创粉丝点击