当前位置: 代码迷 >> 综合 >> leetcode 337 打家劫舍 III
  详细解决方案

leetcode 337 打家劫舍 III

热度:94   发布时间:2023-11-13 12:46:31.0

在这里插入图片描述
这道题真的一开始我想的是树的层次遍历,然后隔行相加,然后再取这二者的最大值。。。

但是如果碰到这个测试用例 [4,1,null,2,null,3],取第1行和第4行的结果为7,而各行求和最大结果才为6

所以就参考了一下leetcode的官方讲解
在这里插入图片描述

//递归的方法
class Solution {private:unordered_map <TreeNode*, int> f, g;//f表示结点选中情况在该点的最大值  g表示结点不选中情况在该点的最大值void dfs( TreeNode* root ){if( !root )return ;dfs( root->left );dfs( root->right );f[root] = root->val + g[root->left] + g[root->right];g[root] = max( f[root->left], g[root->left] ) + max( f[root->right], g[root->right] );}public:int rob(TreeNode* root) {dfs( root );return max( f[root], g[root] );}
};