Codeforces 666E. Forensic Examination

Description

给出串 \(S\) ,和 \(m\) 个串 \(T_i\) ,每次询问 \((l,r,pl,pr)\) 表示 \(S[pl...pr]\)\(T[l...r]\) 中哪一个出现次数最多,求出现次数和编号
题面

Solution

基础题...
对于 \(S,T[l...r]\) 放在一起建广义后缀自动机
然后每次倍增到 S[pl,pr] ,然后查询子树内出现次数最多的 \(T\) 即可
我们可以开一棵 \([1,m]\) 的线段树,维护每一个 \(T\) 的出现次数,维护最大值和最大值位置
线段树合并上来就好了

#include<bits/stdc++.h>
using namespace std;
template<class T>void gi(T &x){
	int f;char c;
	for(f=1,c=getchar();c<'0'||c>'9';c=getchar())if(c=='-')f=-1;
	for(x=0;c<='9'&&c>='0';c=getchar())x=x*10+(c&15);x*=f;
}
const int N=1100010;
int n,m,Q,ch[N][26],cur=1,cnt=1,len[N],fa[N],pos[N],rt[N];char s[N];
int head[N],nxt[N],to[N],num=0,pa[N][20],tt=0,ls[N*20],rs[N*20];
inline void link(int x,int y){nxt[++num]=head[x];to[num]=y;head[x]=num;}
inline void ins(int c){
	int p=cur;cur=++cnt;len[cur]=len[p]+1;
	for(;p && !ch[p][c];p=fa[p])ch[p][c]=cur;
	if(!p)fa[cur]=1;
	else{
		int q=ch[p][c];
		if(len[p]+1==len[q])fa[cur]=q;
		else{
			int nt=++cnt;len[nt]=len[p]+1;
			memcpy(ch[nt],ch[q],sizeof(ch[nt]));
			fa[nt]=fa[q];fa[cur]=fa[q]=nt;
			for(;p && ch[p][c]==q;p=fa[p])ch[p][c]=nt;
		}
	}
}
struct data{int w,p;}tr[N*20];
inline data upd(data x,data y){
	if(x.w>=y.w)return x;
	return y;
}
inline int merge(int x,int y){
	if(!x||!y)return x+y;int o=++tt;
	if(!ls[x] && !rs[x]){
		tr[o].w=tr[x].w+tr[y].w;tr[o].p=tr[x].p;
		return o;
	}
	ls[o]=merge(ls[x],ls[y]);
	rs[o]=merge(rs[x],rs[y]);
	tr[o]=upd(tr[ls[o]],tr[rs[o]]);
	return o;
}
inline void mdf(int &x,int l,int r,int sa){
	if(!x)x=++tt;
	if(l==r){tr[x].w++;tr[x].p=l;return ;}
	int mid=(l+r)>>1;
	if(sa<=mid)mdf(ls[x],l,mid,sa);
	else mdf(rs[x],mid+1,r,sa);
	tr[x]=upd(tr[ls[x]],tr[rs[x]]);
}
inline void dfs(int x){
	for(int i=1;i<=19;i++)pa[x][i]=pa[pa[x][i-1]][i-1];
	for(int i=head[x];i;i=nxt[i])dfs(to[i]),rt[x]=merge(rt[x],rt[to[i]]);
}
inline int lca(int x,int k){
	for(int i=19;i>=0;i--)if(len[pa[x][i]]>=k)x=pa[x][i];
	return x;
}
inline data qry(int x,int l,int r,int sa,int se){
	if(sa<=l && r<=se)return tr[x];
	int mid=(l+r)>>1;
	if(se<=mid)return qry(ls[x],l,mid,sa,se);
	if(sa>mid)return qry(rs[x],mid+1,r,sa,se);
	return upd(qry(ls[x],l,mid,sa,mid),qry(rs[x],mid+1,r,mid+1,se));
}
int main(){
  freopen("pp.in","r",stdin);
  freopen("pp.out","w",stdout);
  scanf("%s",s+1);n=strlen(s+1);
  for(int i=1;i<=n;i++)ins(s[i]-'a'),pos[i]=cur;
  cin>>m;
  for(int i=1;i<=m;i++){
	  scanf("%s",s+1);cur=1;
	  for(int j=1,len=strlen(s+1);j<=len;j++)
		  ins(s[j]-'a'),mdf(rt[cur],1,m,i);
  }
  for(int i=2;i<=cnt;i++)link(fa[i],i),pa[i][0]=fa[i];
  dfs(1);
  int x,y,l,r;
  cin>>Q;
  while(Q--){
	  gi(l);gi(r);gi(x);gi(y);
	  x=lca(pos[y],y-x+1);
	  data t=qry(rt[x],1,m,l,r);
	  if(t.w)printf("%d %d\n",t.p,t.w);
	  else printf("%d 0\n",l);
  }
  return 0;
}

posted @ 2018-07-15 11:58  PIPIBoss  阅读(190)  评论(0编辑  收藏  举报