LeetCode - 213. 打家劫舍 II

213. 打家劫舍 II

你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
LeetCode - 213. 打家劫舍 II_第1张图片
解题思路: 此题是LeetCode-198题的改版,将序列限制为一个环,即首尾相连,那么在这种情况下,我们判定第一个元素是否被选中时,还需要考虑最后一个元素是否被选中的情况,那么我将第1个元素分成选中和不选中两种情况即可转化为LeetCode-198题的解法。

LeetCode - 198. 打家劫舍

198. 打家劫舍

// DP, 相较于昨天的题,不同的是序列是首尾相连的,
// 那么我只需将第一个元素分两种情况即偷或者不偷,
// 用两个DP解即可转化为昨天的题
// Time&Space:O(n)
class Solution {
     
public:
    int rob(vector<int>& nums) {
     
        if (nums.empty()) return 0;
        int n = nums.size();
        vector<int> dp1 = nums, dp2 = nums;
        dp1[0] = 0;
        for (int i = 1; i < n; ++i) {
     
            dp1[i] = max(dp1[i - 1], (i >= 2 ? dp1[i - 2] : 0 )+ nums[i]);

        }
        dp2[0] = nums[0];
        for (int i = 1; i < n; ++i) {
     
            dp2[i] = dp2[i - 1];
            if (i >= 2) {
     
                if (i != n - 1 && dp2[i] < dp2[i - 2] + nums[i]) {
     
                    dp2[i] = dp2[i - 2] + nums[i];
                }
            }
        }
        return max(dp1.back(), dp2.back());
    }
};

你可能感兴趣的:(leetcode,leetcode,动态规划)