Gym 100962J Jimi Hendrix (DFS + 树形dp)

40 篇文章 1 订阅
6 篇文章 0 订阅

大体题意:

给你一颗树,树上的边权是字母,给你一个长度为m 的字符串s,求出树上的两个点,使得这两个点之间的路 构成的字符串 s1,使得s是s1的子字符串(不一定连续),如果多解,任意给出两个节点,否则输出-1 -1?

思路:

比赛没有做出来,之后请教的同学,是一个树形dp。 思路很巧妙!

首先我们任选一个节点作为树根节点(不妨选1号节点吧)

我们可以给每一个节点设置为一个结构体,因为要存储许多信息,有从前往后匹配字符串的长度,和从后往前匹配的字符串长度,和左id,右id

然后从这个树根开始dfs,一直到最深处,在回溯的过程更新信息,如果当前节点u的子节点v 的左长度 + 1位置的s能够符合 边权的话,就要把这个状态向u节点转移,右长度同理。

什么时候合适了呢,就是做长度+右长度  大于等于  s的长度的话,就已经匹配成功了,没必要在回溯递归了,可以return了!

详细见代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn = 500000 + 10;
char s[maxn];
struct Node{
    int l,r;
    int lid,rid;
}dp[maxn];
vector<pair<int,char> > g[maxn];
int ans1,ans2;
bool ok;
int len;
void dfs(int u,int fa){
    if (ok)return;
    for (int i = 0; i < g[u].size(); ++i){
        int v = g[u][i].first;
        if (v == fa)continue;
        dfs(v,u);
        if (ok)return ;
        int L,R;
        if (s[dp[v].l+1] == g[u][i].second)L = dp[v].l + 1;
        else L = dp[v].l;
        if (s[len-dp[v].r] == g[u][i].second)R = dp[v].r + 1;
        else R = dp[v].r;
        if (L + dp[u].r >= len){
            ok = 1;
            ans1 = dp[v].lid;
            ans2 = dp[u].rid;
            return ;
        }
        if (R + dp[u].l >= len){
            ok = 1;
            ans1 = dp[u].lid;
            ans2 = dp[v].rid;
            return ;
        }
        if (L > dp[u].l){
            dp[u].l = L;
            dp[u].lid = dp[v].lid;
        }
        if (R > dp[u].r){
            dp[u].r = R;
            dp[u].rid = dp[v].rid;
        }
    }

}
int main(){
    int n, m;
    while(scanf("%d %d",&n, &m) == 2){
        for (int i = 0; i <= n; ++i)g[i].clear();
        for (int i = 1; i <= n; ++i){
            dp[i].rid = dp[i].lid = i;
            dp[i].l = dp[i].r = 0;
        }
        for (int i = 1; i < n; ++i){
            int u,v;
            char ch;
            scanf("%d %d %c",&u, &v, &ch);
            g[u].push_back(make_pair(v,ch));
            g[v].push_back(make_pair(u,ch));
        }
        scanf("%s",s+1);
        len = m;
        ans1 = ans2 = -1;
        ok = 0;
        dfs(1,-1);
//        for (int i = 1; i <= n; ++i){
//            printf("%d %d %d %d\n",dp[i].l,dp[i].r,dp[i].lid,dp[i].rid);
//        }
        printf("%d %d\n",ans1,ans2);
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
PyCharm是一个集成开发环境(IDE),用于Python编程语言的开发。它提供了许多功能,例如代码编辑、调试和版本控制等。Gym是一个用于开发和比较强化学习算法的开源Python库。它提供了许多预定义的环境,供开发者进行强化学习实验。在安装深度强化学习环境的教程中,引用提到了在Windows 10上安装Anaconda、PyTorch、PyCharm和Gym等软件和包。引用提到了需要下载的软件和包,包括Anaconda、Pycharm、Python、PyTorch、Gym以及Pygame等。而引用中提到了通过pip命令来安装Gym库。综上所述,PyCharm和Gym都是用于Python开发和深度强化学习的工具和库。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *3* [DRL环境安装:Win10+Anaconda+PyTorch+PyCharm+Gym](https://blog.csdn.net/weixin_42480812/article/details/112054592)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *2* [Anaconda+PyCharm+PyTorch+Gym深度强化学习环境搭建 送新手直接送进炼丹炉](https://blog.csdn.net/qq_43674552/article/details/127343863)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值