CF666E Forensic Examination （SAM+線段樹合併）

題面見：https://www.luogu.com.cn/problem/CF666E

 

 

題解

一個裸的SAM+線段樹合併

先把詢問離線，對T建廣義SAM，按串的標號建線段樹，dfs一下跑一遍線段樹合併

在S串中枚舉右端點，如果不存在的話就像kmp一樣調fail指針知道存在，順便記錄一下當前匹配的長度

如果長度不合法（就是詢問串長度大於匹配了的串的長度），說明這個詢問的串是在T中不存在的，直接continue

如果合法就倍增定位當前詢問串的位置，在其對應的線段樹中查詢一下就好了

這SB題我竟然調了一上午，後來發現是insert的時候沒有把x與mid比大小。。。

代碼：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
inline int gi()
{
	char c;int num=0,flg=1;
	while((c=getchar())<'0'||c>'9')if(c=='-')flg=-1;
	while(c>='0'&&c<='9'){num=num*10+c-48;c=getchar();}
	return num*flg;
}
#define N 500005
int n,m,Q;
struct anode{
	int x,id;
	anode(){}
	anode(int a,int b){x=a;id=b;}
	anode operator + (const anode &t)const{
		if(x<t.x)return t;
		if(x>t.x)return *this;
		if(id<t.id)return *this;
		return t;
	}
};
#define lc a[i].l
#define rc a[i].r
struct node{
	int l,r;
	anode x;
}a[N*19];
int tot;
void insert(int &i,int l,int r,int x)
{
	if(!i)i=++tot;
	if(l==r){a[i].x.x++;a[i].x.id=l;return;}
	int mid=(l+r)>>1;
	if(x<=mid)insert(a[i].l,l,mid,x);
	else insert(a[i].r,mid+1,r,x);
	a[i].x=a[lc].x+a[rc].x;
}
int merge(int x,int y,int l,int r)
{
	int i=++tot;
	if(!x||!y){a[i]=a[x+y];return i;}
	int mid=(l+r)>>1;
	if(l==r){
		a[i].x.x=a[x].x.x+a[y].x.x;
		a[i].x.id=l;
		return i;
	}
	a[i].l=merge(a[x].l,a[y].l,l,mid);
	a[i].r=merge(a[x].r,a[y].r,mid+1,r);
	a[i].x=a[lc].x+a[rc].x;
	return i;
}
const int INF=0x3f3f3f3f;
anode query(int i,int l,int r,int ql,int qr)
{
	if(!i||ql>r||qr<l)return anode(0,INF);
	if(ql<=l&&r<=qr)return a[i].x;
	int mid=(l+r)>>1;
	return query(lc,l,mid,ql,qr)+query(rc,mid+1,r,ql,qr);
}
int fa[N],ch[N][26],len[N],T[N],sz,las;
void extend(int x,int id)
{
	int p,np,q,nq;
	p=las;las=np=++sz;
	insert(T[np],1,m,id);
	len[np]=len[p]+1;
	for(;p&&!ch[p][x];p=fa[p])ch[p][x]=np;
	if(!p)fa[np]=1;
	else{
		q=ch[p][x];
		if(len[q]==len[p]+1)fa[np]=q;
		else{
			nq=++sz;
			len[nq]=len[p]+1;
			memcpy(ch[nq],ch[q],sizeof(ch[q]));fa[nq]=fa[q];
			for(;p&&ch[p][x]==q;p=fa[p])ch[p][x]=nq;
			fa[q]=fa[np]=nq;
		}
	}
}
int fir[N],to[N],nxt[N],cnt;
void adde(int a,int b){to[++cnt]=b;nxt[cnt]=fir[a];fir[a]=cnt;}
#define LOG 17
int f[LOG+1][N];
void dfs(int u)
{
	f[0][u]=fa[u];
	for(int i=1;i<=LOG;i++)f[i][u]=f[i-1][f[i-1][u]];
	for(int v,p=fir[u];p;p=nxt[p]){
		v=to[p];dfs(v);
		T[u]=merge(T[u],T[v],1,m);
	}
}
int find(int x,int l)
{
	for(int i=LOG;i>=0;i--)
		if(f[i][x]&&len[f[i][x]]>=l)x=f[i][x];
	return x;
}
vector<int> G[N];
anode ans[N];
struct qnode{
	int l,r,ql,qr;
}q[N];
char s[N],t[N];
int main()
{
	sz=las=1;
	int i,j;
	scanf("%s",s+1);n=strlen(s+1);
	m=gi();
	for(i=1;i<=m;i++){
		scanf("%s",t+1);las=1;
		int le=strlen(t+1);
		for(j=1;j<=le;j++)
			extend(t[j]-'a',i);
	}
	for(i=2;i<=sz;i++)adde(fa[i],i);
	dfs(1);
	Q=gi();
	for(i=1;i<=Q;i++){
		q[i].ql=gi();q[i].qr=gi();q[i].l=gi();q[i].r=gi();
		G[q[i].r].push_back(i);
	}
	int now=1,nowlen=0;
	for(i=1;i<=n;i++){
		int x=s[i]-'a';
		while(now&&!ch[now][x])now=fa[now],nowlen=len[now];
		if(!now){now=1;nowlen=0;continue;}
		now=ch[now][x];nowlen++;
		
		for(j=0;j<(int)G[i].size();j++){
			int id=G[i][j],le=q[id].r-q[id].l+1;
			if(le>nowlen)continue;
			int pos=find(now,le);
			ans[id]=query(T[pos],1,m,q[id].ql,q[id].qr);
		}
	}
	for(i=1;i<=Q;i++){
		if(!ans[i].x)ans[i].id=q[i].ql;
		printf("%d %d\n",ans[i].id,ans[i].x);
	}
}