Featured image of post 213. 打家劫舍 II

213. 打家劫舍 II

题目描述

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

给定一个代表每个房屋存放金额的非负整数数组,计算你 在不触动警报装置的情况下 ,今晚能够偷窃到的最高金额。

示例 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 = [1,2,3]
  • 输出:3

提示:

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

解法一:动态规划

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
func max(nums ...int) int {
    res := 0
    for _, val := range nums {
        if val > res {
            res = val
        }
    }
    return res
}

func rob(nums []int) int {
    if len(nums) == 1 {
        return nums[0]
    }
    robRange := func(start, end int) int {
        n := end - start + 1
        if n == 1 {
            return nums[start]
        }
        prev, cur := nums[start], max(nums[start], nums[start+1])
        for i := 2; i < n; i++ {
            tmp := cur
            cur = max(cur, prev+nums[start+i])
            prev = tmp
        }
        return cur
    }
    return max(robRange(0, len(nums)-2), robRange(1, len(nums)-1))
}
Licensed under CC BY-NC-SA 4.0
最后更新于 2023/06/10 11:58:24
comments powered by Disqus
Built with Hugo
主题 StackJimmy 设计