3172: [Tjoi2013]单词 后缀数组

写了一发后缀数组,二分究竟哪里写错了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;
}

你可能感兴趣的:(3172: [Tjoi2013]单词 后缀数组)