LeetCode 337. 打家劫舍 III

主页有其他数据结构内容（持续更新中）

难度：Medium

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int rob(TreeNode* root) {
        vector<int> result = robTree(root);
        return max(result[0], result[1]);
    }
    
    //  返回长度为2的数组，数组下标0表示不偷当前节点时能获取的最大金额，数组下标1表示偷当前节点时能获取的最大金额
    vector<int> robTree(TreeNode* cur) {
        if (cur == NULL) {
            return vector<int>{0, 0};
        }
        //  递归求得当前节点的左右孩子的对应数组
        vector<int> left = robTree(cur->left);
        vector<int> right = robTree(cur->right);
        //  不偷cur，那么可以偷也可以不偷左右节点，因此取较大的情况即可
        int val1 = max(left[0], left[1]) + max(right[0], right[1]);
        //  偷cur，那么就不能偷左右节点
        int val2 = cur->val + left[0] + right[0];
        return {val1, val2};
    }
};