题意:
给一颗边上有字符串的树。求U,V节点U和节点V之间的路径上有多少字符串以S为前缀。
题解:
挺无脑的一道题。虽然想了一会
直接就按跟到节点的路径建可持久化trie就没了。
code:
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;
struct node{
int x,y,next;
char s[12];
}a[200010];int len,last[100010];
struct trnode{
int a[26],c;
}tr[1000005];int tot=0,root[100010];
int fa[100010][20],dep[100010];
int n,m;
void ins(int x,int y,char *s)
{
a[++len].y=y;strcpy(a[len].s,s);
a[len].next=last[x];last[x]=len;
}
int solve(int x,int y)
{
if(dep[x]<dep[y]) swap(x,y);
for(int i=17;i>=0;i--)
if((1<<i)<=dep[x]-dep[y]) x=fa[x][i];
if(x==y) return x;
for(int i=17;i>=0;i--)
if((1<<i)<=dep[x]&&fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
return fa[x][0];
}
void insert(int &x,int froot,int i,int len,char *s)
{
x=++tot;tr[x]=tr[froot];
tr[x].c++;
if(i==len) return;
int c=s[i]-'a';
insert(tr[x].a[c],tr[froot].a[c],i+1,len,s);
}
void dfs(int x,int f)
{
dep[x]=dep[f]+1;fa[x][0]=f;
for(int i=1;(1<<i)<=dep[x];i++)
fa[x][i]=fa[fa[x][i-1]][i-1];
for(int i=last[x];i;i=a[i].next)
{
int y=a[i].y;
if(y==f) continue;
insert(root[y],root[x],0,strlen(a[i].s),a[i].s);
dfs(y,x);
}
}
char s[15];
int get(int x,int i,int len,char *s)
{
if(!x) return 0;
if(i==len) return tr[x].c;
int c=s[i]-'a';
return get(tr[x].a[c],i+1,len,s);
}
int main()
{
scanf("%d",&n);
for(int i=1;i<n;i++)
{
int x,y;char s[12];
scanf("%d %d %s",&x,&y,s);
ins(x,y,s);ins(y,x,s);
}
dep[0]=-1;dfs(1,0);
scanf("%d",&m);
while(m--)
{
int x,y;scanf("%d %d",&x,&y);
int lca=solve(x,y);
scanf("%s",s);
printf("%d\n",get(root[x],0,strlen(s),s)+get(root[y],0,strlen(s),s)-2*get(root[lca],0,strlen(s),s));
}
}