bzoj 4477: [Jsoi2015]字符串树

题意：

给一颗边上有字符串的树。求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));
    }
}