198.打家劫舍 (Easy)*

题目描述*

你是一个专业的小偷，计划偷窃沿街的房屋。每间房内都藏有一定的现金，影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。

给定一个代表每个房屋存放金额的非负整数

示例*

输入: [1,2,3,1]

输出: 4

输入: [2,7,9,3,1]

输出: 12

代码*

简单的动态规划，不过这里如果 dp 数组从小到大还需要处理开头，可以反向 dp。

dp反向 dp

class Solution {
public:
    int rob(vector<int>& nums) {
        vector<int> dp = nums;
        int res = 0;
        if(nums.size() == 0) {
            return 0;
        }else if(nums.size() == 1) {
            return nums[0];
        }else if(nums.size() == 2){
            return max(nums[0], nums[1]);
        }
        dp[0] = nums[0];
        dp[1] = max(nums[1], nums[0]);
        for(int i = 2; i < nums.size(); i++) {
            dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);
            cout << dp[i] << " " ;
            res = max(dp[i], res);
        }
        return res;
    }
};

class Solution {
public:
    int rob(vector<int>& nums) {
        // vector<int> dp(nums.size() + 2);
        int i_1 = 0, i_2 = 0, dp_i = 0;
        for(int i = nums.size() - 1; i >= 0; i--) {
            // dp[i] = max(dp[i + 1], nums[i] + dp[i + 2]);
            dp_i = max(i_1, nums[i] + i_2);
            i_2 = i_1;
            i_1 = dp_i;
        }
        return dp_i;
    }
};

最后更新: July 23, 2022