LeetCode-热题100：198. 打家劫舍

题目描述

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

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

示例 1：

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

示例 2：

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

提示：

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

代码及注释

func rob(nums []int) int {
    // 初始化结果为0
    res := 0
    // 初始化动态规划数组
    dp := make([]int, len(nums))
    
    // 遍历所有房屋
    for i := 0; i < len(nums); i++ {
        if i < 2 {
            // 前两个房屋，只能偷一个
            dp[i] = nums[i]
        } else if i == 2 {
            // 第三个房屋，可以选择偷第一个房屋或第三个房屋
            dp[i] = nums[i] + nums[0]
        } else {
            // 对于每个房屋，可以选择偷当前房屋与前两个房屋的和，或者偷当前房屋与前三个房屋的和
            dp[i] = max(dp[i - 2], dp[i - 3]) + nums[i]
        }
        // 更新最大金额
        res = max(res, dp[i])
    }
    
    // 返回最大金额
    return res
}

// 辅助函数，返回两个数的最大值
func max(a, b int) int {
    if a > b {
        return a
    }
    return b
}

代码解释

这里使用了动态规划的方法：

• 定义 dp[i] 为偷到第 i 个房屋时的最大金额。
• 对于前两个房屋，直接将 dp[i] 设为 nums[i]。
• 对于第三个房屋，因为不能连续偷两个房屋，所以 dp[i] 可以是 nums[i] + nums[0] 。
• 对于后续的房屋，选择偷当前房屋与前两个房屋的和，或者偷当前房屋与前三个房屋的和的最大值。

最后，返回 dp 数组中的最大值作为结果。