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。

相关推荐
飞滕人生TYF10 小时前
动态规划 详解
算法·动态规划
SSL_lwz14 小时前
P11290 【MX-S6-T2】「KDOI-11」飞船
c++·学习·算法·动态规划
ZZZ_O^O16 小时前
【动态规划-卡特兰数——96.不同的二叉搜索树】
c++·学习·算法·leetcode·动态规划
橘子遇见BUG1 天前
算法日记 31 day 动态规划(01背包)
算法·动态规划
我是哈哈hh1 天前
专题二十二_动态规划_子序列系列问题(数组中不连续的一段)_算法专题详细总结
数据结构·c++·算法·动态规划·子序列
橘子遇见BUG2 天前
算法日记 30 day 动态规划(背包问题)
算法·动态规划
是糖不是唐2 天前
代码随想录算法训练营第五十二天|Day52 图论
c语言·算法·深度优先·动态规划·图论
Lindsey_feiren2 天前
代码随想录day44算法随想录|动态规划07
算法·动态规划
姜西西_2 天前
动态规划 之 子序列系列 算法专题
算法·动态规划
丶Darling.2 天前
Day46 | 动态规划 :线性DP 最长递增子序列&&最长连续递增子序列
算法·动态规划