●BZOJ 3172 [Tjoi2013]单词

Posted 2020-10-16 *ZJ

题链：

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;
}