LeetCode #337 - House Robber III

题目描述:

The thief has found himself a new place for his thievery again. There is only one entrance to this area, called the "root." Besides the root, each house has one and only one parent house. After a tour, the smart thief realized that "all houses in this place forms a binary tree". It will automatically contact the police if two directly-linked houses were broken into on the same night.

Determine the maximum amount of money the thief can rob tonight without alerting the police.

Example 1:

     3
    / \
   2   3
    \   \ 
     3   1
Maximum amount of money the thief can rob =  3  +  3  +  1  =  7 .

Example 2:

     3
    / \
   4   5
  / \   \ 
 1   3   1

Maximum amount of money the thief can rob = 4 + 5 = 9.

给定一个二叉树,只能抢劫不相邻的节点,求最大的金额。方法是采用递归,但是由于不一定经过根结点,所以分经过根结点和不经过根结点两种情况计算,然后求最大值,也就是每次递归都返回两个值,下一次的递推也根据这两个值继续进行。

class Solution {
public:
    int rob(TreeNode* root) {
        pair<int,int> result=rob_node(root);
        return max(result.first,result.second);
    }
    
    pair<int,int> rob_node(TreeNode* root)
    {
        pair<int,int> result(0,0);
        if(root==NULL) return result;
        pair<int,int> rob_left=rob_node(root->left);
        pair<int,int> rob_right=rob_node(root->right);
        result.first=root->val+rob_left.second+rob_right.second;
        result.second=max(rob_left.first,rob_left.second)+max(rob_right.first,rob_right.second);
        return result;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值