P6088-[JSOI2015]字符串树【可持久化Trie,LCA】

正题

题面链接:https://www.luogu.com.cn/problem/P6088


题目大意

n n n个点的一棵树,每条边上有一个字符串,求一条路径上有多少以询问字符串为前缀的字符串。


解题思路

建立一个可持久化 T r i e Trie Trie,然后每个点继承父亲插入字符串。

然后询问答案就是 s u m x + s u m y − s u m L C A ( x , y ) sum_x+sum_y-sum_{LCA(x,y)} sumx+sumysumLCA(x,y)

时间复杂度 O ( n log ⁡ n ) O(n\log n) O(nlogn)


c o d e code code

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e5+10,T=17;
struct node{
	int to,next,w;
}a[N*2];
int n,m,cnt,tot,rt[N],dep[N],ls[N];
int f[N][T+1],ch[N*11][26],w[N*11];
char s[N][11],qr[11];
void addl(int x,int y,int w){
	a[++tot].to=y;
	a[tot].next=ls[x];
	ls[x]=tot;a[tot].w=w;
	return;
}
int Insert(int x,int k,int id,int lim){
	int y=++cnt;w[y]=w[x]+1;
	if(k>=lim)return y;
	memcpy(ch[y],ch[x],sizeof(ch[y]));
	int c=s[id][k]-'a';
	ch[y][c]=Insert(ch[x][c],k+1,id,lim);
	return y;
}
int Ask(int x,int lim){
	for(int i=0;i<lim;i++)
		x=ch[x][qr[i]-'a'];
	return w[x];
}
void dfs(int x,int fa){
	dep[x]=dep[fa]+1;
	for(int i=ls[x];i;i=a[i].next){
		int y=a[i].to;
		if(y==fa)continue;
		rt[y]=Insert(rt[x],0,a[i].w,strlen(s[a[i].w]));
		f[y][0]=x;dfs(y,x);
	}
	return;
}
int LCA(int x,int y){
	if(dep[x]>dep[y])swap(x,y);
	for(int i=T;i>=0;i--)
		if(dep[f[y][i]]>=dep[x])y=f[y][i];
	if(x==y)return x;
	for(int i=T;i>=0;i--)
		if(f[x][i]!=f[y][i])
			x=f[x][i],y=f[y][i];
	return f[x][0];
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		int x,y;
		scanf("%d%d%s",&x,&y,s[i]);
		addl(x,y,i);addl(y,x,i);
	}
	dfs(1,1);
	for(int j=1;j<=T;j++)
		for(int i=1;i<=n;i++)
			f[i][j]=f[f[i][j-1]][j-1];
	scanf("%d",&m);
	while(m--){
		int x,y,lim;
		scanf("%d%d",&x,&y);
		scanf("%s",qr);lim=strlen(qr);
		printf("%d\n",Ask(rt[x],lim)+Ask(rt[y],lim)-2*Ask(rt[LCA(x,y)],lim));
	}
	return 0;
}
已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 精致技术 设计师:CSDN官方博客 返回首页