1. 程式人生 > >●BZOJ 3172 [Tjoi2013]單詞

●BZOJ 3172 [Tjoi2013]單詞

pac scan har ios cst itl cstring std break

題鏈:

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

●BZOJ 3172 [Tjoi2013]單詞