深度优先遍历--怎么抓住小偷

时间:2022-07-23
本文章向大家介绍深度优先遍历--怎么抓住小偷，主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项，具有一定的参考价值，需要的朋友可以参考一下。

问题描述：每栋别墅以二叉树的结构建立，每个节点的值是财富值，小偷不能够偷相接的别墅，问小偷最多能够偷多少财富值?

有个规律：每一个节点的偷值都是：左侧子节点的不偷值+右侧子节点的不偷值+该节点的值；不偷值：左侧子节点的最大值+右侧子节点的最大值。我们可以先深度优先遍历到每一个叶子节点，再依此计算。

class TreeNode:
    de __init(self,x,left=None,right=None):
        self.val=x
        self.left=left
        self.right=right
t3=TreeNode(1);t4=TreeNode(3);t6=TreeNode(1)
t1=TreeNode(4,t3,t4);t2=TreeNode(5,left=None,right=t6)
root=TreeNode(3,t1,t2)

def rob(root):
    a=helper(root)
    return max(a[0],a[1])
def helper(root):
    if root == None:
        return (0,0)
    left=helper(root.left)
    right=helper(root.right)
    robval=rot.val+left[1]+right[1]
    skipval=max(left[0],left[1])+max(right[0],right[1])
    return (robval,skipval)
输出：9

主要考察建模能力以及实现能力。

随机文章
本站知识点必读
最近更新