如果偷了父节点,就不能动子节点。
要找最大值,遍历不可避免,第一反应是回溯法,不过因为树是无序的,并不存在剪枝规则,所以回溯其实就是不停的递归。
- 根节点被偷,子节点必然不能偷;
- 根节点没被偷,子节点可偷可不偷。
暴力递归
参考这个题解:
三种方法解决树形动态规划问题-从入门级代码到高效树形动态规划代码实现 - 打家劫舍 III - 力扣(LeetCode)
爷爷,儿子,孙子的例子非常形象。
代码还是有点抽象的,而且会超时
class Solution {
public:
int rob(TreeNode* root) {
if(!root) return 0;
int money = root->val;
if(root->left)
money += rob(root->left->left) + rob(root->left->right);
if(root->right)
money += rob(root->right->left) + rob(root->right->right);
return max(money, rob(root->left) + rob(root->right));
}
};
简单优化
既然是重复计算,那就可以一遍计算一遍记录嘛,计算之前查表看之前有没有计算过,如果就算过直接查表获取值就可以了。
class Solution {
public:
unordered_map<TreeNode*, int> map;
int rob(TreeNode* root) {
if(!root) return 0;
if(!root->left && !root->right) return root->val;
if(map[root]) return map[root];
int money1 = rob(root->left) + rob(root->right);//不偷父节点
//偷父节点,该父节点的子节点就不能偷了
int money2 = root->val;
if(root->left) money2 += rob(root->left->left) + rob(root->left->right);
if(root->right) money2 += rob(root->right->left) + rob(root->right->right);
map[root] = max(money1, money2);
return map[root];
}
};