题链:
http://www.lydsy.com/JudgeOnline/problem.php?id=3172
题解:
把单词逐个接起来,中间用互不相同的字符连接,并记录下每个单词的首字母在串中的位置。
(因为太多,所以转成int数组,用大数连接)
然后计算出 sa[],rak[],hei[]数组。
并求出 L[i]: 找出排名最小(即为 L[i])的后缀与排名为 i的后缀的 LCP==hei[i]
同理R[i]:找出排名最大(即为 R[i])的后缀与排名为 i的后缀的 LCP==hei[i]
可以用单调栈 O(n)维护。
然后对于每个单词,只要它在后缀数组中与它前面一个后缀或后面一个后缀的LCP==len,
那么答案就是 R[]-L[]+1,否则就为 1。
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#define MAXN 1005000
#define filein(x) freopen(#x".in","r",stdin);
#define fileout(x) freopen(#x".out","w",stdout);
using namespace std;
char S[MAXN];
int A[MAXN],P[MAXN],sa[MAXN],rak[MAXN],hei[MAXN],L[MAXN],R[MAXN];
void build(int N,int M){
static int cc[MAXN],ta[MAXN],tb[MAXN],*x,*y,h,p;
x=ta; y=tb; h=0;
for(int i=0;i<M;i++) cc[i]=0;
for(int i=0;i<N;i++) cc[x[i]=A[i]]++;
for(int i=1;i<M;i++) cc[i]+=cc[i-1];
for(int i=N-1;i>=0;i--) sa[--cc[x[i]]]=i;
for(int k=1;p=0,k<N;k<<=1){
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;
for(int i=0;i<M;i++) cc[i]=0;
for(int i=0;i<N;i++) cc[x[y[i]]]++;
for(int i=1;i<M;i++) cc[i]+=cc[i-1];
for(int i=N-1;i>=0;i--) sa[--cc[x[y[i]]]]=y[i];
swap(x,y); y[N]=-1; x[sa[0]]=0; M=1;
for(int i=1;i<N;i++)
x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?M-1:M++;
if(M>=N) break;
}
for(int i=0;i<N;i++) rak[sa[i]]=i; A[N]=-1;
for(int i=0,j;i<N;i++){
if(h) h--;
if(rak[i]){
j=sa[rak[i]-1];
while(A[i+h]==A[j+h]) h++;
}
hei[rak[i]]=h;
}
}
void pre(int N){
static int stk[MAXN],stp[MAXN],top;
top=0; stp[top]=0;
for(int i=0;i<N;i++){
while(top&&stk[top]>=hei[i]) top--;
L[i]=stp[top]; top++;
stk[top]=hei[i]; stp[top]=i;
}
top=N; stp[top]=N;
for(int i=N-1;i>=0;i--){
while(top<N&&stk[top]>=hei[i]) top++;
R[i]=stp[top]-1; top--;
stk[top]=hei[i]; stp[top]=i;
}
}
int main()
{
int len=0,N; scanf("%d",&N);
for(int i=1;i<=N;i++){
P[i]=len;
scanf("%s",S+len);
len=strlen(S);
S[len++]='&'; S[len]=0;
}
P[N+1]=len=strlen(S);
for(int i=0,cnt=10;i<len;i++) A[i]=S[i]=='&'?MAXN-cnt++:S[i];
/*
for(int i=1;i<=N;i++) printf("%d ",P[i]); printf("\n");
for(int i=0;i<len;i++) printf("%c ",S[i]); printf("\n");
for(int i=0;i<len;i++) printf("%d ",A[i]); printf("\n");*/
build(len,MAXN-5);
pre(len);
for(int i=1,ANS,r,l;i<=N;i++){
r=rak[P[i]]; l=P[i+1]-P[i]-1;
if(r==0||hei[r+1]==l) r++;
if(hei[r]!=l) ANS=1;
else ANS=R[r]-L[r]+1;
printf("%d\n",ANS);
}
return 0;
}