这道题真的一开始我想的是树的层次遍历,然后隔行相加,然后再取这二者的最大值。。。
但是如果碰到这个测试用例 [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] );}
};