【20230314】【每日一题】打家劫舍III

在上次打劫完一条街道之后和一圈房屋后，小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为“根”。 除了“根”之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫，房屋将自动报警。

计算在不触动警报的情况下，小偷一晚能够盗取的最高金额。

---

树形dp

1. 确定递归函数的参数和返回值

一个节点偷与不偷的两个状态，返回值是一个长度为2的数组

dp数组及其下标的含义：下标为0记录不偷该节点所得最大金钱，下标为1记录偷该节点所得最大金钱。

2. 确定终止条件

if(node==NULL) return vector<int> {0,0};

3. 确定遍历顺序

后序遍历 因为通过递归函数的返回值来做下一步计算

4. 确定单层递归逻辑

如果不偷当前节点，左右孩子可以考虑被偷：

int val0=max(left[0],left[1])+max(right[0],right[1]);

如果偷当前节点，左右孩子不能偷：

int val1=node->val+left[0]+right[0];

最后当前节点的状态就是 {val0,val1}

class Solution {
public:
    vector<int> robber(TreeNode* node){
        if(node==NULL)  return vector<int> {0,0};
        //后序遍历
        vector<int> left=robber(node->left);
        vector<int> right=robber(node->right);
        //单层处理逻辑
        //如果不偷当前节点，那么就考虑偷当前节点的左右孩子
        int val0=max(left[0],left[1])+max(right[0],right[1]);
        //如果偷当前节点,那么就必定偷不了左右孩子
        int val1=node->val+left[0]+right[0];
        return {val0,val1};
    }
    int rob(TreeNode* root) {
        vector<int> result=robber(root);
        return max(result[0],result[1]);
    }
};