第一道树状DP。
第一种方法,也是最容易想到的基本的dp。一个节点的最大钱数,就是opt(left)+opt(right)和val+ll+lr+rl+rr的最大值。这样做的话,会增加很多搜索的时间,时间1000ms左右
/**
* 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) {
return opt(root);
}
int opt(TreeNode* node){
if(!node) return 0;
TreeNode* lt=node->left;
TreeNode* rt=node->right;
int ll=0,lr=0,rr=0,rl=0;
if(lt) {
ll=opt(lt->left);
lr=opt(lt->right);
}
if(rt) {
rl=opt(rt->left);
rr=opt(rt->right);
}
int l=opt(lt);
int r=opt(rt);
return (node->val+ll+lr+rr+rl)>(l+r)?(node->val+ll+lr+rr+rl):l+r;
}
};
而另一种方法,核心思想:每个节点有2种状态,选这个节点或者不选这个节点。然后用一个vector储存起来(0为选,1为不选)。那么,这个节点的最大值为left[1]+right[1]+val和max(left)+max(right)中的最大值。这样只用dfs两边,而不是刚刚的6个点。
/**
* 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:
vector<int> dfs(TreeNode* node){
if(!node) return vector<int> (2,0);
vector<int> lf=dfs(node->left);
vector<int> rt=dfs(node->right);
vector<int> anss(2,0);
anss[0]=lf[1]+rt[1]+node->val;//要这个点
anss[1]=max(lf[0],lf[1])+max(rt[0],rt[1]);
return anss;
}
int rob(TreeNode* root) {
vector<int> ans=dfs(root);
return max(ans[0],ans[1]);
}
};