【JSOI2015】字符串树

Description

$1\leq n,Q\leq 100000$，且输入所有字符串长度不超过$10$
n为树中节点数，Q为询问数

Analysis

之前做的题，忘记写题解了QAQ
Ans(u,v)=Ans(root,u)+Ans(root,v)-2*Ans(root,lca(u,v))
以s为前缀的串的判定，可以想到哈希、Trie
在这道题用Trie就比较简单啦，儿子以父亲的信息，边上的串来建可持久化Trie（也就是对于每个点都建一个Trie，重复的信息用指针指一下）
那么查询就直接在点的Trie上跑一遍就好了

Code

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,b,a) for(int i=b;i>=a;i--)
#define efo(i,v) for(int i=last[v];i;i=next[i])
using namespace std;
const int N=100005,M=N*2;
int n,tot,num,len,to[M],next[M],lth[M],s[M][12],last[N],f[N][20],dep[N],root[N];
struct node
{
    int x,son[27];
}a[N*11];
char st[12];
void link(int u,int v)
{
    to[++tot]=v,next[tot]=last[u],last[u]=tot;
    lth[tot]=strlen(st+1);
    fo(i,1,lth[tot]) s[tot][i]=st[i]-'a';
}
void build(int u,int v,int p)
{
    fo(i,1,lth[p])
    {
        a[v]=a[u];
        a[v].son[s[p][i]]=++num,a[v].x++;
        u=a[u].son[s[p][i]],v=a[v].son[s[p][i]];
    }
    a[v].x++;
}
void dfs(int v,int fr,int d)
{
    f[v][0]=fr,dep[v]=d;
    efo(i,v)
    {
        int u=to[i];
        if(u==fr) continue;
        build(root[v],root[u],i);
        dfs(u,v,d+1);
    }
}
int getlca(int u,int v)
{
    if(dep[u]<dep[v]) swap(u,v);
    fd(i,log2(dep[u]),0)
        if(dep[f[u][i]]>=dep[v]) u=f[u][i];
    if(u==v) return u;
    fd(i,log2(dep[u]),0)
        if(f[u][i]!=f[v][i]) u=f[u][i],v=f[v][i];
    return f[u][0];
}
int calc(int v)
{
    v=root[v];
    fo(i,1,len)
    {
        v=a[v].son[st[i]];
        if(!v) return 0;
    }
    return a[v].x;
}
int main()
{
    freopen("strings.in","r",stdin);
    freopen("strings.out","w",stdout);
    int u,v;
    scanf("%d",&n);
    fo(i,1,n-1)
    {
        scanf("%d %d %s\n",&u,&v,st+1);
        link(u,v),link(v,u);
    }
    fo(i,1,n) root[i]=++num;
    dfs(1,1,1);
    fo(j,1,log2(n))
        fo(i,1,n) f[i][j]=f[f[i][j-1]][j-1];
    int _;
    for(scanf("%d",&_);_;_--)
    {
        scanf("%d %d %s\n",&u,&v,st+1);
        len=strlen(st+1);
        fo(i,1,len) st[i]=st[i]-'a';
        int lca=getlca(u,v);
        printf("%d\n",calc(u)+calc(v)-2*calc(lca));
    }
    return 0;
}