二叉树:抢劫房间

    若干房子组成一棵二叉树,每个房间放有固定的金钱,小偷抢劫房子时,如果抢劫连续的两间房子就会触发报警。求小偷最多能抢到多少金钱。

    分析:采用递归的方法,访问每个节点时返回两个值:一个抢劫当前房间时,当前房间和其左右子树能抢到的金钱数;另一个不抢劫当前房间时,左右子树能抢到的金钱数。

public class Solution {
    public int rob(TreeNode root) {
        int[] num = dfs(root);
        return Math.max(num[0], num[1]);
    }
    private int[] dfs(TreeNode x) {
        if (x == null) return new int[2];
        int[] left = dfs(x.left);
        int[] right = dfs(x.right);
        int[] res = new int[2];
        res[0] = left[1] + right[1] + x.val;//抢劫当前房间,其左右子节点则不能抢劫
        res[1] = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);//不抢劫当前房间,其左右子节点可抢也可不抢,取其最大值
        return res;
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值