leetcode-hot100 1. 打家劫舍III (cpp)

图片来自leetcode 截图

//对于相邻的两层节点，第一层右边的节点和第二层左边的节点完全可以求和的!!所以不能用层次遍历求和来迁移到一维DP计算
//树状DP 自底向上
#include<iostream>
#include<unordered_map>
#include<algorithm>
using namespace std;

struct TreeNode 
{
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode(int x): val(x),left(NULL),right(NULL){}
};

//递归-记忆化(使用map存储已经计算的子节点，直接返回值）
class Solution {
public:
	unordered_map<TreeNode*, int> memo;
	
	int rob(TreeNode* root) {
		return dorob(root);
	}

	//计算每一个以当前节点为根节点的子树的最大值
	int dorob(TreeNode* root)
	{
		if (root == NULL)
			return 0;
		if (memo.count(root))
			return memo[root];
		int res1=0, res2=0;
		
		//计算当前节点
		if (root->left)
		{
			res1 += dorob(root->left->right) + dorob(root->left->left);
		}
		if (root->right)
		{
			res1 += dorob(root->right->left) + dorob(root->right->right);
		}		
		res1 += root->val;
	
		//不计算当前节点
		res2 = dorob(root->left) + dorob(root->right);

		memo[root] = max(res1, res2);
		return memo[root];
	}

};