213. 打家劫舍 II

Difficulty: 中等

你是一個專業的小偷，計劃偷竊沿街的房屋，每間房內都藏有一定的現金。這個地方所有的房屋都 圍成一圈 ，這意味著第一個房屋和最後一個房屋是緊挨著的。同時，相鄰的房屋裝有相互連通的防盜系統，如果兩間相鄰的房屋在同一晚上被小偷闖入，系統會自動報警 。

給定一個代表每個房屋存放金額的非負整數陣列，計算你 在不觸動警報裝置的情況下 ，能夠偷竊到的最高金額。

示例1：

輸入：nums = [2,3,2]
輸出：3
解釋：你不能先偷竊 1 號房屋（金額 = 2），然後偷竊 3 號房屋（金額 = 2）, 因為他們是相鄰的。

示例 2：

輸入：nums = [1,2,3,1]
輸出：4
解釋：你可以先偷竊 1 號房屋（金額 = 1），然後偷竊 3 號房屋（金額 = 3）。
    偷竊到的最高金額 = 1 + 3 = 4 。
 

示例 3：

輸入：nums = [0]
輸出：0

提示：

• 1 <= nums.length <= 100
• 0 <= nums[i] <= 1000

Solution

Language: ****

這道題是LeetCode 198. 打家劫舍 - swordspoet - 部落格園的一個變化，只是題目說明首尾兩家是相鄰的，不能同時搶，但是可以搶一家，所以當只有兩家的時候不是一家都不可以搶的，所以我們可以把搶劫分成兩種情況，一種只含有除最後一家，一種只含有除第一家，然後分別搶就可以了，最後取max。還有一種特殊情況就是隻有一家的情況下直接返回就可以了，第一次提交沒有注意這個情況提交錯誤。

class Solution:
  def rob(self, nums: List[int]) -> int:
    # 首位不能同時搶，那麼把首尾分開，分成兩種情況討論
    if not nums:
      return 0
    if len(nums) == 1:
      return nums[0]
    nums1, nums2 = nums[1:], nums[:-1]
    return max(self.robHelper(nums1), self.robHelper(nums2))
    
  def robHelper(self, nums):
    dp = [0] * (len(nums) + 1)
    dp[0], dp[1] = 0, nums[0]
    
    for i in range(2, len(nums)+1):
      dp[i] = max(dp[i-2] + nums[i-1], dp[i-1])
      
    return dp[len(nums)]