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