动态十八：打家劫舍 III

题目地址： https://leetcode-cn.com/problems/house-robber-iii/
题目描述：
在上次打劫完一条街道之后和一圈房屋后，小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为“根”。 除了“根”之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫，房屋将自动报警。
计算在不触动警报的情况下，小偷一晚能够盗取的最高金额。

68747470733a2f2f696d672d626c6f672e6373646e696d672e636e2f32303231303232333137333834393631392e706e67.png

参考代码：

class Solution {
public:
    int rob(TreeNode* root) {
        vector<int> result =  getResult(root);
        return max(result[0], result[1]);
    }
    
    
//    dp[root] 代表 以root 为 根结点到 目标值
    vector<int> getResult(TreeNode* root) {
        vector<int> result = vector<int>(2,0);
        if (root == nullptr) {
            result[0] = 0; // 没有选
            result[1] = 0; // 选root
            return result;
        }
        vector<int> left = getResult(root->left);
        vector<int> right = getResult(root->right);
        // 选中 root
        result[1] = root->val+left[0]+right[0];
        // 没有选root
        result[0] = max(left[0],left[1]) + max(right[0], right[1]);
        return result;
    }
};

参考链接： https://github.com/youngyangyang04/leetcode-master/blob/master/problems/0337.%E6%89%93%E5%AE%B6%E5%8A%AB%E8%88%8DIII.md