本文最后更新于598 天前,其中的信息可能已经过时,如有错误请发送邮件到tomding1065@gmail.com
视频讲解:https://www.bilibili.com/video/BV1H24y1Q7sY
https://programmercarl.com/0337.%E6%89%93%E5%AE%B6%E5%8A%AB%E8%88%8DIII.html
1.本题是作为树形dp的入门题,跟着卡哥系统的过了一遍二叉树章节之后来做这个就没有什么难度了,本题也是考虑到后序遍历的操作,我觉得不仅是本题,动态规划的题里最难的就是对于dp数组的定义问题,这个真的是一个很难想到的一个点,尤其是对于这个题我尤其的赞同这个观点,本题的dp数组的定义是只有两个数dp[0]:不偷该节点的数值的最大金钱数。dp[1]:偷该节点的数值的最大金钱数。然后就是进行后序遍历,根据递归遍历二叉树,携带本层遍历到的这个节点的dp的状态,实现状态转移,求解结果。
CPP
/**
* 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:
vector<int> robTree(TreeNode* cur){
if(cur == NULL)return vector<int> {0,0};
vector<int> leftdp = robTree(cur->left);
vector<int> rightdp = robTree(cur->right);
int val0 = max(leftdp[0], leftdp[1]) + max(rightdp[0], rightdp[1]);
int val1 = cur->val + leftdp[0] + rightdp[0];
return vector<int>{val0, val1};
}
int rob(TreeNode* root) {
vector<int> dp = robTree(root);
return max(dp[0], dp[1]);
}
};