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