Featured image of post Leetcode-198 打家劫舍

Leetcode-198 打家劫舍

题目

你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。

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

示例 1:

1
2
3
4
输入:[1,2,3,1]
输出:4
解释:偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
偷窃到的最高金额 = 1 + 3 = 4 。

示例 2:

1
2
3
4
输入:[2,7,9,3,1]
输出:12
解释:偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。
偷窃到的最高金额 = 2 + 9 + 1 = 12 。

提示:

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

递归实现

实际上与爬楼梯是同一道题,解法和思路都一致

每个房屋(第 n 个)均要做一下判断:

  • 偷:那么偷到的总金额 = 当前房屋的金额 + 在第 n-2 个房屋和之前偷到的最大金额,既 amount(n) + f(n-2)
  • 不偷:那么偷到的总金额 = 在第 n-1 个房屋和之前偷到的最大金额,既 f(n-1)
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
func robRecursion(nums []int) int {
    var dfs func(int) int
    dfs = func(i int) int {
        if i < 0 {
            return 0
        }
        return max(dfs(i-1), dfs(i-2)+nums[i])
    }

    return dfs(len(nums) - 1)
}

递归 + 记忆化

省略

自底向上递推

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
func rob(nums []int) int {
    var (
        prev1 = 0 // 对应 i-1
        prev2 = 0 // 对应 i-2
    )
    for _, n := range nums {
        prev2, prev1 = prev1, max(prev1, prev2+n)
    }
    return prev1
}