●BZOJ 3879 SvT

题链:

http://www.lydsy.com/JudgeOnline/problem.php?id=3879

题解:

后缀数组,单调栈,RMQ
其实类似 BZOJ 3238 [Ahoi2013]差异
只是不过变成了多次询问某些后缀两两之间的LCP的和。
所以对于每次询问,就先把那些后缀提出来,
按 rank排序后,求出相邻的后缀的 LCP(存储在 H 数组中),
然后再按那个题的做法做就好了(即求出 H 数组对应的 L[],R[])。

读入数据有点大,用了读入优化。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 505000
#define filein(x) freopen(#x".in","r",stdin);
#define fileout(x) freopen(#x".out","w",stdout);
using namespace std;
char S[MAXN];
int	sa[MAXN],rak[MAXN],hei[MAXN],L[MAXN],R[MAXN],stm[MAXN][20],log2[MAXN];
char gc(){
	//return getchar();
	static char IN[MAXN];
	static int bit=MAXN-5000,p=0,len=0;
	if(p>=len) len=fread(IN,1,bit,stdin),IN[len]=EOF,p=0; 
	return IN[p++];
}
void read(int &x){
	static int f;
	static char ch;
	x=0; f=1; ch=gc(); 
	while(ch<'0'||'9'<ch){if(ch=='-')f=-1;ch=gc();} 
	while('0'<=ch&&ch<='9'){x=x*10+ch-'0';ch=gc();} 
	x=x*f;
}
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]=S[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;
	for(int i=0,j;i<N;i++){
		if(h) h--;
		if(rak[i]){
			j=sa[rak[i]-1];
			while(S[i+h]==S[j+h]) h++;
		}
		stm[rak[i]][0]=hei[rak[i]]=h;
	}
	for(int k=1;k<=log2[N];k++)
		for(int i=(1<<k)-1;i<N;i++)
			stm[i][k]=min(stm[i-(1<<(k-1))][k-1],stm[i][k-1]);
}
int LCP(int l,int r){
	static int k;
	if(l>r) swap(l,r); l++;
	k=log2[r-l+1];
	return min(stm[l+(1<<k)-1][k],stm[r][k]);
}
void preLR(int N,int *val){
	static int stk[MAXN],stp[MAXN],top;
	stp[top=0]=0;
	for(int i=0;i<N;i++){
		while(top&&stk[top]>=val[i]) top--;
		L[i]=stp[top]; top++;
		stk[top]=val[i]; stp[top]=i;
	}
	stp[top=N]=N;
	for(int i=N-1;i>=0;i--){
		while(top<N&&stk[top]>val[i]) top++;
		R[i]=stp[top]-1; top--;
		stk[top]=val[i]; stp[top]=i;
	}
}
int main()
{
	//filein(3879);
	long long ans;
	log2[1]=0; for(int i=2;i<=500000;i++) log2[i]=log2[i>>1]+1;
	static int A[MAXN*10],H[MAXN*10],N,M; 
	read(N); read(M);
	for(int i=0;i<N;i++) 
		do{S[i]=gc();}while(S[i]<'a'||'z'<S[i]);
	S[N]=0;build(N,300);
	for(int i=0,t;i<M;i++){
		ans=0;
		read(t); for(int j=0;j<t;j++) read(A[j]),A[j]=rak[A[j]-1];
		sort(A,A+t);
		t=unique(A,A+t)-A;
		for(int i=1;i<t;i++) H[i]=LCP(A[i-1],A[i]);
		preLR(t,H);
		for(int i=1;i<t;i++)
			ans+=1ll*(i-L[i])*(R[i]-i+1)*H[i];
		printf("%lld\n",ans);
	}
	return 0;
}

posted @ 2017-12-07 20:08  *ZJ  阅读(160)  评论(0编辑  收藏  举报