由于字符串长度不超过10,所以trie的深度就不超过10。
对于一条边u,v,s
那么我们将v从u上继承信息,然后对于v的trie树,插入s这个串,将路径上每一个点的权值都+1
不过我们不可能对每一个点都开一个trie,
我们会发现每一次加入一条边,只有字符串长度个点发生改变,
所以,就用可持久化trie。
#include
#include
#include
#include
#include
#define N 100003
using namespace std;
struct arr
{
int x,son[26];
}t[10*N];
int a[N*2],b[N],next[N*2],v[N*2],tot,len[N],f[N][20];
char s[N][20],st[20];
int n,m,x,y,ans,l,root[N],deep[N];
void ins(int x,int y,int z)
{
next[++tot]=b[x];
a[tot]=y;
v[tot]=z;
b[x]=tot;
}
void in(int x,int fa,int z)
{
root[x]=++tot;
fa=root[fa];
for(int i=0;i'a']=++tot;
fa=t[fa].son[s[z][i]-'a'];
}
t[tot].x++;
}
void dfs(int x,int fa)
{
for(int i=b[x];i;i=next[i])
if(a[i]!=fa)
{
in(a[i],x,v[i]);
f[a[i]][0]=x;
deep[a[i]]=deep[x]+1;
dfs(a[i],x);
}
}
int lca(int x,int y)
{
if(deep[x]for(int j=16;j>=0;j--)
if(deep[f[x][j]]>=deep[y])x=f[x][j];
for(int j=16;j>=0;j--)
if(f[x][j]!=f[y][j])x=f[x][j],y=f[y][j];
if(x==y)return x;else return f[x][0];
}
int find(int x,int i)
{
if(i==l)return t[x].x;
return find(t[x].son[st[i]-'a'],i+1);
}
int main()
{
freopen("strings.in","r",stdin);
freopen("strings.out","w",stdout);
scanf("%d",&n);
for(int i=1;iscanf("%d%d%s",&x,&y,&s[i]);
ins(x,y,i);
ins(y,x,i);
len[i]=strlen(s[i]);
}
deep[1]=tot=root[1]=1;
dfs(1,0);
for(int j=1;j<17;j++)
for(int i=1;i<=n;i++)
f[i][j]=f[f[i][j-1]][j-1];
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d%s",&x,&y,&st);
l=strlen(st);
printf("%d\n",find(root[x],0)+find(root[y],0)-2*find(root[lca(x,y)],0));
}
}