题目:
小偷又发现了一个新的可行窃的地区。这个地区只有一个入口,我们称之为 root 。
除了 root 之外,每栋房子有且只有一个“父“房子与之相连。一番侦察之后,聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果 两个直接相连的房子在同一天晚上被打劫 ,房屋将自动报警。
给定二叉树的 root 。返回 在不触动警报的情况下 ,小偷能够盗取的最高金额 。
解析:
单纯一个递归题目而已,不是dp。首先想到当前节点的最大值的两种产生情况。然后就想如果dp那么dp数组下标代表什么,应该用什么,迷茫了。单纯递归就好,然后写完代码提交超时了,然后发现计算当前节点会计算其左右孩子,左右孙子的最大值。那么当前节点为孩子时即会将原当前节点的孙子重新计算一遍造成超时。所以可以使用一个map把计算过的结果保存一下,这样如果计算过孙子了,那么计算孩子的时候可以复用孙子节点的结果
class Solution {
public:
unordered_map<TreeNode* , int> umap; // 记录计算过的结果
int rob(TreeNode* root) {
if (umap[root]) return umap[root];
if(!root) return 0;
if(!root->left && !root->right) return root->val;
int val1 = root->val;
if(root->left) val1 += rob(root->left->left)+rob(root->left->right);
if(root->right) val1 += rob(root->right->left)+rob(root->right->right);
int val2 = rob(root->left) +rob(root->right);
umap[root] = max(val1,val2);
return max(val1,val2);
}
};