Leetcode 打家劫舍 III

第一道树状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]);
    }
};