LeetCode - 解题笔记 - 213 - House Robber II

Solution 1

0198. House Robber 的变体，额外增加了一个条件：当抢第一个房子时，最后一个房子不能抢。

基于此，需要同时判断两个初始值不同的策略链：其中一个抢第一个房子（那么第二个就不能抢），那么最佳结果要考察倒数第二个状态；其中一个不抢第一个房子（那么第二个房子一定要抢），最佳结果要考察倒数第一个状态。

• 时间复杂度：$O(N)$，其中$N$为输入房子的个数，线性遍历一次
• 空间复杂度：$O(1)$，仅维护常数个状态量

class Solution {
public:
    int rob(vector<int>& nums) {
        if (nums.size() <= 1) {
            return nums.empty()? 0: nums[0];
        }
        
        // 方案1：抢第一个房子，那么最后一个房子就不能抢
        int first1 = nums[0], second1 = nums[0];
        // 方案2：不抢第一个房子，那么最后一个房子可以抢
        int first2 = 0, second2 = nums[1];
        
        for (int i = 2; i < nums.size(); ++i) {
            int temp1 = second1, temp2 = second2;
            second1 = max(second1, first1 + nums[i]);
            second2 = max(second2, first2 + nums[i]);
            
            first1 = temp1;
            first2 = temp2;
        }
        
        int ans = max(first1, second2);
        
        return ans;
    }
};

Solution 2

Solution 1的Python实现

class Solution:
    def rob(self, nums: List[int]) -> int:
        if len(nums) <= 1:
            return 0 if len(nums) == 0 else nums[0]
        
        first1, second1 = nums[0], nums[0]
        first2, second2 = 0, nums[1]
        
        for i in range(2, len(nums)):
            temp1, temp2 = second1, second2
            
            second1 = max(second1, first1 + nums[i])
            second2 = max(second2, first2 + nums[i])
            
            first1, first2 = temp1, temp2
        
        ans = max(first1, second2)
        
        return ans