题意分析:给定一棵二叉树,求能获取的权值最大和(相邻的不能同时取)
解题思路:
方法1:利用哈希表和递归回溯实现
当前的计算需要依赖之前的结果,对于某一个节点,如果其左子节点存在,通过递归调用函数计算不包含坐姿节点的返回值,如果右子节点存在,通过递归调用函数,算法不包含右子节点的返回值,对应此节点的最大值存在两种情况:一是该节点值加上不包含左子节点和左子节点的返回值之和,另一种是左右子节点返回值之和不包含当前节点值,去两者的较大值返回即可。但是递归调用中,存在重复计算,故可以利用hash表存储节点,来加快执行速度。具体实现见下面程序。
方法2:递归实现
递归递推式为:
rob_root = max(rob_L + rob_R, no_rob_L + no_rob_R + root.val);
no_rob_root = rob_L + rob_R
C++实现
// 方法1实现 int helper(TreeNode *root, map<TreeNode *, int> &mp) { if (!root) return 0; if (mp.count(root)) return mp[root]; int val = 0; if (root->left) { val += helper(root->left->left, mp) + helper(root->left->right, mp); } if (root->right) { val += helper(root->right->left, mp) + helper(root->right->right, mp); } val = max(val + root->val, helper(root->left, mp) + helper(root->right, mp)); mp[root] = val; return val; } // 递归求解 int rob(TreeNode *root) { map<TreeNode *, int> mp; return helper(root, mp); } //方法2实现 pair<int, int> dfs(TreeNode *root) { pair<int, int> dp = make_pair(0, 0); if (root) { pair<int, int> dp_L = dfs(root->left); pair<int, int> dp_R = dfs(root->right); dp.second = dp_L.first + dp_R.first; dp.first = max(dp.second, dp_L.second + dp_R.second + root->val); } return dp; } int rob1(struct TreeNode *root) { return dfs(root).first; }
参考文献
leetcode_337 House Robber III
最新推荐文章于 2021-05-27 23:36:09 发布