主页有其他数据结构内容(持续更新中)
难度: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};
}
};