【Lintcode】535. House Robber III

题目地址:

https://www.lintcode.com/problem/house-robber-iii/

给定一棵二叉树,要求选取一些节点使得任意两个节点不相邻并且节点总和最大。求那个最大值。

这是个树上DP问题。设树根为 x x x,若取 x x x则最大和为 f [ x ] f[x] f[x],若不取 x x x则最大和为 g [ x ] g[x] g[x],显然想要求的答案就是 max ⁡ { f [ x ] , g [ x ] } \max\{f[x],g[x]\} max{f[x],g[x]}。而 f [ x ] = x . v + g [ x . l ] + g [ x . r ] g [ x ] = max ⁡ { f [ x . l ] , g [ x . l ] } + max ⁡ { f [ x . r ] , g [ x . r ] } f[x]=x.v+g[x.l]+g[x.r]\\g[x]=\max\{f[x.l],g[x.l]\}+\max\{f[x.r],g[x.r]\} f[x]=x.v+g[x.l]+g[x.r]g[x]=max{f[x.l],g[x.l]}+max{f[x.r],g[x.r]}只需要DFS一遍就可以得到。代码如下:

public class Solution {
    /**
     * @param root: The root of binary tree.
     * @return: The maximum amount of money you can rob tonight
     */
    public int houseRobber3(TreeNode root) {
        // write your code here
        int[] dp = dfs(root);
        return Math.max(dp[0], dp[1]);
    }
    
    // 返回一个长度为2的数组,第一个数字是不取root所得最大和,第二个数字是取root所得最大和
    private int[] dfs(TreeNode root) {
        if (root == null) {
            return new int[]{0, 0};
        }
        
        int[] left = dfs(root.left), right = dfs(root.right);
        int[] dp = {0, 0};
        
        dp[0] = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
        dp[1] = root.val + left[0] + right[0];
        
        return dp;
    }
}

class TreeNode {
    int val;
    TreeNode left, right;
    
    public TreeNode(int val) {
        this.val = val;
    }
}

时间复杂度 O ( n ) O(n) O(n),空间 O ( h ) O(h) O(h)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值