动态规划---打家劫舍(3)

题目:

小偷又发现了一个新的可行窃的地区。这个地区只有一个入口,我们称之为 root 。

除了 root 之外,每栋房子有且只有一个“父“房子与之相连。一番侦察之后,聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果 两个直接相连的房子在同一天晚上被打劫 ,房屋将自动报警。

给定二叉树的 root 。返回 在不触动警报的情况下 ,小偷能够盗取的最高金额 。

思路:

本题涉及到二叉树,处理二叉树的方法必不可少的就是递归与回溯,接下来结合递归三部曲与动态规划五部曲来综合分析问题。

1.确定递归函数的参数与返回值

每一个节点有偷和不偷两种状态,那么递归函数的返回值就是一个长度为2的int数组,其实这里返回的数组就是dp数组,dp数组的第一个元素记录不偷该节点可以得到的金钱值,第二个元素记录偷该节点可以得到的金钱值。

2.确定终止条件

遇到空节点就返回[0,0]

3.确定遍历顺序

递归方法使用后序遍历,因为要通过递归函数的返回值来做下一步计算。

通过递归左节点,得到左节点偷与不偷的金钱

通过递归右节点,得到右节点偷与不偷的金钱

4.确定单层递归逻辑

如果不偷当前节点,那么左右孩子就可以偷,选一个最大的偷。

val1=max(left[0],left[1])+max(right[0],right[1])

如果偷当前节点,那么左右孩子就不能偷

val2=left[0]+right[0]+node.val

最后当前节点的状态就是[val1,val2]

代码:

    public int rob(TreeNode root) {
        int[] result=robAction(root);
        return Math.max(result[0],result[1]);
    }
    public int[] robAction(TreeNode root){
        int[] res=new int[2];
        if(root==null) return res;
        //递归
        int[] left=robAction(root.left);
        int[] right=robAction(root.right);
        //处理
        res[0]=Math.max(left[0],left[1])+Math.max(right[0],right[1]);
        res[1]=root.val+left[0]+right[0];
        return res;
    }

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值