【树形DP+DFS】Gym 100962J - Jimi Hendrix

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/monochrome00/article/details/83041800

题目链接<https://cn.vjudge.net/problem/360051/origin>

---

题意：

给出一个树，树上边的权值是字母。询问一个字符串，问能否在路上找到一条简单路径使它的子序列包含该字符串。输出任意的简单路径的两个端点即可。

---

题解：

如果一个节点，它的一个子树能够从左到右匹配到第ll个字符，另外一个子树能从右往左匹配到第rr个字符，而且ll+1>=rr那么就代表成功匹配。

每个节点记录四个量，从左到右最多匹配到第几个字符，以及初始的端点。从右到左最多匹配到第几个字符，以及初始的端点。

注意：一定是两个不同的子树才能做判断。

---

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+7;
int n,m;
struct Edge{
    int u,v,w,nxt;
    Edge(int u=0,int v=0,int w=0,int nxt=0):u(u),v(v),w(w),nxt(nxt){}
}e[N*2];
int p[N],edn;
void add(int u,int v,int w){
    e[++edn]=Edge(u,v,w,p[u]);p[u]=edn;
    e[++edn]=Edge(v,u,w,p[v]);p[v]=edn;
}
struct Node{
    int li,ri;
    int ll,rl;
}dp[N];
char s[N];
int a,b;
bool dfs(int u,int fa){
    dp[u].li=dp[u].ri=u;
    dp[u].ll=0,dp[u].rl=m+1;
    for(int i=p[u];~i;i=e[i].nxt){
        int v=e[i].v;
        if(v==fa) continue;
        if(dfs(v,u)) return true;
        
        int tl=dp[v].ll;
        if(e[i].w==s[dp[v].ll+1]) tl++;
        if(tl+1>=dp[u].rl){
            a=dp[v].li;
            b=dp[u].ri;
            return true;
        }

        int tr=dp[v].rl;
        if(e[i].w==s[dp[v].rl-1]) tr--;
        if(dp[u].ll+1>=tr){
            a=dp[u].li;
            b=dp[v].ri;
            return true;
        }

        if(tr<dp[u].rl){
            dp[u].rl=tr;
            dp[u].ri=dp[v].ri;
        }
        if(tl>dp[u].ll){
            dp[u].ll=tl;
            dp[u].li=dp[v].li;
        }
    }
    return false;
}
char w[3];
int main()
{
    scanf("%d%d",&n,&m);
    memset(p,-1,sizeof(p));edn=-1;
    for(int i=1;i<=n;i++){
        int u,v;
        scanf("%d%d%s",&u,&v,&w);
        add(u,v,w[0]);
    }
    scanf("%s",s+1);
    a=b=-1;
    dfs(1,0);
    /*for(int i=1;i<=n;i++){
        printf("%d %d\n",dp[i].ll,dp[i].rl);
    }*/
    printf("%d %d\n",a,b);
}