leetcode 337 打家劫舍 III

这道题真的一开始我想的是树的层次遍历，然后隔行相加，然后再取这二者的最大值。。。

但是如果碰到这个测试用例 [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] );

    }
};