bzoj4477 [Jsoi2015]字符串树

Description

---

萌萌买了一颗字符串树的种子，春天种下去以后夏天就能长出一棵很大的字
符串树。字符串树很奇特，树枝上都密密麻麻写满了字符串，看上去很复杂的样
子。

问题描述

字符串树本质上还是一棵树，即N个节点N-1条边的连通无向无环图，节点
从1到N编号。与普通的树不同的是，树上的每条边都对应了一个字符串。萌萌
和JYY在树下玩的时候，萌萌决定考一考JYY。每次萌萌都写出一个字符串S和
两个节点U,V，需要JYY立即回答U和V之间的最短路径（即,之间边数最少的
路径。由于给定的是一棵树，这样的路径是唯一的）上有多少个字符串以为前
缀。
JYY虽然精通编程，但对字符串处理却不在行。所以他请你帮他解决萌萌的难题。

输入中所有字符串只包含a-z的小写字母。
1<=N,Q<=100,000，且输入所有字符串长度不超过10。

Solution

---

很容易想到求一次lca，令ans[x]为x到根的答案，那么答案为ans[x]+ans[y]-ans[lca]*2
这样dfs一次，每个点以父亲为上一个版本建trie，查询的时候直接找就可以了

空间不太够，那么就可持久化一下

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)

const int N=100505;
const int L=15;

int root[N],count[N*L];
int rec[N*L][26],tot;
int fa[N][21],dep[N];
int ls[N],edCnt;
char str[N*2][L],opt[L];

struct edge {int y,next;} e[N*2];

void addEdge(int x,int y) {
    e[++edCnt]=(edge) {y,ls[x]}; ls[x]=edCnt;
    scanf("%s",str[edCnt]);
    e[++edCnt]=(edge) {x,ls[y]}; ls[y]=edCnt;
    memcpy(str[edCnt],str[edCnt-1],sizeof(str[edCnt]));
}

void insert(int pre,int &y,char *str) {
    int len=strlen(str);
    int now=y=++tot;
    rep(i,0,len-1) {
        memcpy(rec[now],rec[pre],sizeof(rec[now]));
        count[now]=count[pre]+1;
        rec[now][str[i]-'a']=++tot;
        pre=rec[pre][str[i]-'a'];
        now=rec[now][str[i]-'a'];
    }
    count[now]=count[pre]+1;
}

int query(int now,char *str) {
    int len=strlen(str);
    rep(i,0,len-1) {
        now=rec[now][str[i]-'a'];
    }
    return count[now];
}

int get_lca(int x,int y) {
    if (dep[x]<dep[y]) std:: swap(x,y);
    drp(i,20,0) if (dep[fa[x][i]]>=dep[y]) x=fa[x][i];
    if (x==y) return x;
    drp(i,20,0) if (fa[x][i]!=fa[y][i]) {
        x=fa[x][i];
        y=fa[y][i];
    }
    return fa[x][0];
}

void dfs(int now) {
    rep(i,1,20) fa[now][i]=fa[fa[now][i-1]][i-1];
    for (int i=ls[now];i;i=e[i].next) {
        if (e[i].y==fa[now][0]) continue;
        insert(root[now],root[e[i].y],str[i]);
        dep[e[i].y]=dep[now]+1;
        fa[e[i].y][0]=now;
        dfs(e[i].y);
    }
}

int main(void) {
    root[1]=tot=1;
    int n; scanf("%d",&n);
    rep(i,2,n) {
        int x,y; scanf("%d%d",&x,&y);
        addEdge(x,y);
    }
    dep[1]=1; dfs(1);
    int T; scanf("%d",&T);
    while (T--) {
        int x,y; scanf("%d%d",&x,&y);
        scanf("%s",opt);
        int lca=get_lca(x,y);
        int ans=query(root[x],opt)+query(root[y],opt);
        ans-=query(root[lca],opt)*2;
        printf("%d\n",ans);
    }
    return 0;
}