【leetcode】337. 打家劫舍 III

题目描述：337

解题思路：打家劫舍本质是01背包问题，关注点在于：当前节点值的取舍、对于节点上一状态即左右子节点的状态值。

当前状态值（当前值+不包含左右子节点的最大状态值，左右子节点最大状态值之和）。

class Solution {
public:
    int rob(TreeNode* root) {
        vector<int> res = dfs(root);
        return max(res[0], res[1]);
    }
    vector<int> dfs(TreeNode* root){
        vector<int> res(2,0);
        if(!root) return res;
        vector<int> l = dfs(root->left);
        vector<int> r = dfs(root->right);
        res[0] = max(l[0],l[1]) + max(r[0], r[1]);//不包含当前root的最大值，左子节点两种情况的较大值加上右子节点两种情况的较大值
        res[1] = root->val + l[0] + r[0];//不包含左子节点值的最大值加上不包含右子节点值的最大值，和当前节点值之和
        return res;
    }
};
//写成pair
class Solution {
public:
    int rob(TreeNode* root) {
         pair<int,int> res = dfs(root);
        return max(res.second, res.first);
    }
    pair<int,int> dfs(TreeNode* root){
        pair<int,int> res(0,0);
        if(!root) return res;
         pair<int,int> l = dfs(root->left);
         pair<int,int> r= dfs(root->right);
        res.first = max(l.first , l.second) + max(r.first, r.second);
        res.second = root->val + l.first  + r.first ;
        return res;
    }
};