leetcode打家劫舍(数组/环状排列/二叉树)--想哭的动态规划

打家劫舍1 数组

你是一个专业的小偷，计划偷窃沿街的房屋。每间房内都藏有一定的现金，影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。给定一个代表每个房屋存放金额的非负整数数组，计算你 不触动警报装置的情况下 ，一夜之内能够偷窃到的最高金额。
来源：力扣（LeetCode）
链接：打家劫舍
题记
简单来说就是打劫一排住户,但不能打劫邻近的住户,求打劫最大值
定义dp[]数组用来存储最大抢劫量,其中dp[i]用来表示抢劫到第i个住户时的最大量,相邻的两家不能抢,所以如果抢了i-1那么i户就不能抢了,即
dp[i]=max{dp[i-1],dp[i-2]+nums[i]}
初始状态：
前 0 间房子的最大偷窃价值为 0，即 dp[0] = 0。
返回值：
返回 dp列表最后一个元素值，即所有房间的最大偷窃价值。
简化空间复杂度：
我们发现 dp[n] 只与 dp[n-1]和 dp[n-2]有关系，因此我们可以设两个变量 cur和 pre 交替记录，将空间复杂度降到 O(1) 。

class Solution {
public:
    int rob(vector<int>& nums) {
        if(nums.size()==0) return 0;
        int pre1=0;
        int pre2=0;
        for(int i=0; i<nums.size(); i++){
            int cur=max(pre2+nums[i],pre1);
            pre2=pre1;
            pre1=cur;
        }
        return pre1;

    }
};

打家劫舍2 环状排列

你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈，这意味着第一个房屋和最后一个房屋是紧挨着的。同时，相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组，计算你在不触动警报装置的情况下，能够偷窃到的最高金额。
来源：力扣（LeetCode）
链接：打家劫舍2
题记
上一个是一排,这一个是首尾相接.也就是说抢了第一间就不能抢最后一间,还是假设dp[i]是打劫到i住户时的最大价值
不抢劫第一间:最大值是nums[1:]即从nums[1]家开始.最大金额是p1;
抢劫第一间不抢最后一间:最大值是nums[:n-1],即从第0家开始到倒数第二家结束.最大金额是p2;
最大金额价值即max{p1,p2}.

class Solution {
public:
    int rob(vector<int>& nums) {
        int n=nums.size();
        if (n==0) return 0;
        if (n==1) return nums[0];
        return max(rob_1(nums,1,n),rob_1(nums,0,n-1));
    }
    int rob_1(vector<int>& nums, int start, int end){
        int pre1=0,pre2=0,cur=0;
        for(int i=start; i<end; i++){
            cur = max(pre2+nums[i],pre1);
            pre2 = pre1;
            pre1 = cur;
        }
        return pre1;
    }
};

打家劫舍3 二叉树

在上次打劫完一条街道之后和一圈房屋后，小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为“根”。 除了“根”之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫，房屋将自动报警。
计算在不触动警报的情况下，小偷一晚能够盗取的最高金额。
链接：打家劫舍3
题记
一棵二叉树，树上的每个点都有对应的权值，每个点有两种状态（选中和不选中），问在不能同时选中有父子关系的点的情况下，能选中的点的最大权值和是多少。f(o) 表示选择 o 节点的情况下，oo节点的子树上被选择的节点的最大权值和；g(o)表示不选择 o 节点的情况下，o节点的子树上被选择的节点的最大权值和；l 和 r代表 o的左右孩子。二叉树结构与前面两个不同的是左子树和右子树要遍历
当o选中,o的左右孩子不能选:f(o)=g(l)+g[r]
当o不选,o的左右孩子可以选也可以不选:g(o)=max(f(l),g(l))+max(f[r],g[r]), 根节点的f和g的最大值就是答案

/**
 * 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:
    unordered_map<TreeNode*,int>f,g;
    void dfs(TreeNode* o){
        if(!o){
            return;
        }
        dfs(o->left);
        dfs(o->right);
        //选该节点后该节点子树最大价值和
        f[o]=o->val+g[o->left]+g[o->right];
        //不选该节点,该节点的子树既可以被选也可以不被选
        g[o]=max(f[o->left],g[o->left])+max(f[o->right],g[o->right]);
        //f[]和g[]都是存放的值,哈希表理解为字典
    }
    int rob(TreeNode* root) {
        dfs(root);
        return max(f[root],g[root]);

    }
};