题目链接<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);
}