棋子

导航

House Robber III leetcode 动态规划

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;
};

 

posted on 2016-03-13 00:19  鼬与轮回  阅读(1272)  评论(0编辑  收藏  举报