力扣 198. 打家劫舍

题目来源:https://leetcode.cn/problems/house-robber/description/

力扣 198. 打家劫舍_第1张图片

 C++题解:因为是间接偷窃,所以偷nums[i]家前,一定偷过第i-2或者i-3家,因为i-1不能偷。

例如12345共5家,先偷第1家,那么2不能偷,如果偷5那肯定偷3,所以只在3和4之间作判断就行。即跳跃为2或3步。dp[i] = max(dp[i-2]+nums[i-1], dp[i-3]+nums[i-1]);

class Solution {
public:
    int rob(vector& nums) {
        int len = nums.size();
        if(len == 1) return nums[0];
        vector dp(len+1, 0); // 在前i家偷到的最大金额
        dp[0] = 0;
        dp[1] = nums[0];
        dp[2] = nums[1];
        for(int i = 3; i <= len; i++) {
            dp[i] = max(dp[i-2]+nums[i-1], dp[i-3]+nums[i-1]);
        }
        return max(dp[len], dp[len-1]);
    }
};

代码随想录题解:

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

你可能感兴趣的:(开始C++吧,leetcode,算法,c++,动态规划)