Leetcode 2920. Maximum Points After Collecting Coins From All Nodes

  • [Leetcode 2920. Maximum Points After Collecting Coins From All Nodes](#Leetcode 2920. Maximum Points After Collecting Coins From All Nodes)
    • [1. 解题思路](#1. 解题思路)
    • [2. 代码实现](#2. 代码实现)

1. 解题思路

这一题思路上也很直接,就是一个深度优先遍历加上一个动态规划,分别考虑在两种情况下对应的coin收集数目,然后取最大值即可。

2. 代码实现

给出python代码实现如下:

python 复制代码
class Solution:
    def maximumPoints(self, edges: List[List[int]], coins: List[int], k: int) -> int:
        graph = defaultdict(list)
        for u, v in edges:
            graph[u].append(v)
            graph[v].append(u)
            
        @lru_cache(None)
        def dp(u, p, m):
            if m >= 14:
                return 0
            val = coins[u] 
            for _ in range(m):
                val = val // 2
            ans = val - k
            for v in graph[u]:
                if v == p:
                    continue
                ans += dp(v, u, m)
            if val - k < val // 2:
                ans2 = val // 2
                for v in graph[u]:
                    if v == p:
                        continue
                    ans2 += dp(v, u, m+1)
                ans = max(ans, ans2)
            return ans
        
        ans = dp(0, -1, 0)
        return ans

提交代码评测得到:耗时3186ms,占用内存424.7MB。

相关推荐
Perishell2 小时前
无人机避障——大疆与Airsim中的角速度信息订阅获取
linux·动态规划·无人机
yannan201903139 小时前
【算法】(Python)动态规划
python·算法·动态规划
姜西西_1 天前
动态规划 之 斐波那契数列模型 算法专题
算法·动态规划
丶Darling.2 天前
代码随想录 | Day38 | 动态规划 :01背包应用 目标和&&一和零
c++·算法·动态规划·记忆化搜索·回溯·递推·01背包
tan77º2 天前
【AcWing】算法基础课-动态规划
算法·动态规划·1024程序员节
陈琦煜2 天前
AcWing 1073 树的中心 树形dp (详解)
动态规划·树形dp
£suPerpanda2 天前
牛客周赛 Round65 补题DEF
开发语言·数据结构·c++·算法·深度优先·动态规划·图论
hxj..2 天前
【算法】动态规划
java·算法·动态规划
waterme1onY3 天前
算法笔记:Day-09(初始动态规划)
java·笔记·算法·动态规划
十七算法实验室4 天前
Matlab实现海洋捕食者优化算法(MPA)求解路径规划问题
开发语言·算法·决策树·支持向量机·matlab·动态规划·启发式算法