leetcode337. 打家劫舍 III

一:题目

二:上码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    /**
    思路:
        1.动态规划YYDS
        2.分析题意:既然是二叉树  那么和他离不开的就是遍历顺序  深度遍历和广度遍历  我们可以知道的是本题和父节点
          有关的，那么我们就不考虑广度遍历;我们是考虑子节点的状态来考虑我们是否偷取父节点;所以我们采用后序遍历。
        3.用到递归
            1>:确定递归函数与参数
                我们采取两个状态来记录我们是否偷取 0 表示不偷; 1 表示偷;

                vector<int> sumMoney(TreeNode* root)

            2>:确定递归的终止条件
                当我们遍历到叶节点的时候  又往下遍历了一次 那么我们的结点返回必定是不偷
                if(root == NULL) return {0,0}
            3>:确定递归的递归体
                vector<int> left = sumMoney(root->left);
                vector<int> right = sumMoney(root->right);

                那么我们的递归体就是为了表达当前结点偷与不偷;

                如果我们偷当前的结点;那么我们的当前已偷的总价值就是加上左右节不偷的价值
                int val1 = cur->val + left[0] + right[0];

                如果我们不偷当前的结点;那么我们现在的总价值就是就是左右结点中偷与不偷中价值的最大价值
                int val2 = max(left[0],left[1]]) + max(right[0],right[1])


    **/

    int rob(TreeNode* root) {
        vector<int> ans;
        ans =  sumMoney(root);
        return max(ans[0],ans[1]);
    }

     vector<int> sumMoney(TreeNode* root) {
        if(root == NULL) return {0,0};
        
        vector<int> left = sumMoney(root->left);
        vector<int> right = sumMoney(root->right);

        int val1 = root->val + left[0] + right[0];//偷
        int val2 = max(left[0],left[1]) + max(right[0],right[1]);//不偷

        return {val2,val1};//注意这里的顺序我们是 {不偷,偷} 因为我们的下标安排的是0 1 下标就是不偷与偷
     }
};

天天向上 加油