您的位置:首页 > 其它

House Robber III leetcode 动态规划

2016-03-13 00:19 471 查看
https://leetcode.com/submissions/detail/56095603/

这是一道不错的DP题!自己想了好久没有清晰的思路,参看大神博客!http://siukwan.sinaapp.com/?p=1013

我自己的感悟:

对于树的的前序遍历(根-左-右);编写代码时,有效代码(处理函数)写在递归左子树之后,递归右子树之前;

中序遍历(左-根-右)用于二叉排序树的顺序输出,编写代码时,有效代码(处理函数)写在递归左右子树之前;

后序遍历(左-右-根),有效代码写在递归左右子树之后;

本题就是采用的后序遍历,对于每一个节点,先处理好其左右子树之后,再考虑本身节点的情况,非常符合实际:

/**
* Definition for a binary tree node.
* struct TreeNode {
*     int val;
*     TreeNode *left;
*     TreeNode *right;
*     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int rob(TreeNode* root) {
if(root==NULL)
return 0;
dfs(root);
return max(dp1[root],dp2[root]);
}
void dfs(TreeNode* node){
if(node==NULL)
return;
dfs(node->left);
dfs(node->right);//实际干活的代码写在这两个递归之后,表示先处理一个节点的左右子树之后,才考虑本身节点(后序遍历处理手法)
dp1[node]=dp2[node->left]+dp2[node->right]+node->val;//dp2[NULL]是等于0的,注意,只有key是指针类型才有效,本题满足条件
dp2[node]=max(max(dp1[node->left]+dp1[node->right],dp2[node->left]+dp2[node->right]),max(dp1[node->left]+dp2[node->right],dp1[node->right]+dp2[node->left]));
}
private:
map<TreeNode*,int> dp1;
map<TreeNode*,int> dp2;
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: