【JSOI2015】字符串树

这里写图片描述

题解

由于字符串长度不超过10,所以trie的深度就不超过10。
对于一条边u,v,s
那么我们将v从u上继承信息,然后对于v的trie树,插入s这个串,将路径上每一个点的权值都+1
不过我们不可能对每一个点都开一个trie,
我们会发现每一次加入一条边,只有字符串长度个点发生改变,
所以,就用可持久化trie。

code

#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));
    }
}

你可能感兴趣的:(题解,trie)