6、网络升级改造

最新推荐文章于 2024-07-25 11:27:31 发布

Python ml

最新推荐文章于 2024-07-25 11:27:31 发布

阅读量102

点赞数 2

分类专栏： python 文章标签：深度优先算法

本文链接：https://blog.csdn.net/m0_48711099/article/details/137052165

版权

python 专栏收录该内容

66 篇文章 0 订阅

订阅专栏

该篇文章介绍了一种通过动态规划和深度优先搜索算法解决的问题，给定一个树形结构和每个节点的成本，目标是找到删除节点以使得剩余子树中所有节点的最大成本最小。通过递归计算不同策略下的成本，最终确定最小总成本。

摘要由CSDN通过智能技术生成

n = int(input())
nums = list(map(int,input().split()))

dp = {}
def dfs(nodeIdx):
    if nodeIdx>=n:return 0
    if nodeIdx in dp: return dp[nodeIdx]
    # 选择去除当前节点，则左孩子，右孩子不能去掉，只能去掉他们的孩子
    l,r = 2*nodeIdx + 1,2*nodeIdx + 2
    ll,lr = 2*l+1, 2*l+2
    rl,rr = 2*r+1, 2*r+2
    cost = nums[nodeIdx] + dfs(ll)+dfs(lr)+dfs(rl)+dfs(rr)
    # 不去除当前节点，则左孩子，右孩子都能去掉
    cost = max(cost, dfs(l)+dfs(r))
    dp[nodeIdx] = cost
    return cost

print(dfs(0))