打家劫舍

LeetCode 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 。

思路

相邻不能同时偷，所以走到第 (i) 家只有两档：不偷它，沿用 dp[i-1]；偷它，就只能加上 dp[i-2]。取大的。

1. dp[i]：前 i+1 家能偷到的最高金额。
2. dp[i] = max(dp[i-1], dp[i-2] + nums[i])。
3. dp[0] = nums[0]；dp[1] = max(nums[0], nums[1])。空数组返回 0。
4. 空间能压成两个变量。时间 (O(n))。

[2,1,1,2] 这种两端都大的，第二家和第三家都得放下，答案是 2+2=4，不是顺着偷过去。

顺便画一个比较简单的易懂的图示：

假设输入为 [2, 7, 9, 3, 1]，动态规划表如下：

i	0	1	2	3	4
nums[i]	2	7	9	3	1
dp[i]	2	7	11	11	12

• 对于第 2 个房屋（索引为 1），可以选择偷第一个房屋或第二个房屋，最大金额为 7。
• 对于第 3 个房屋（索引为 2），可以选择偷第一个和第三个房屋，最大金额为 11。
• 对于第 4 个房屋（索引为 3），可以选择偷第二个和第四个房屋，最大金额为 11。
• 对于第 5 个房屋（索引为 4），可以选择偷第一个、第三个和第五个房屋，最大金额为 12。

参考代码

C++

#include <iostream>
#include <vector>
#include <algorithm>

int rob(const std::vector<int>& nums) {
    int n = nums.size();
    if (n == 0) return 0;
    if (n == 1) return nums[0];
    
    // 初始化 dp 数组
    std::vector<int> dp(n);
    dp[0] = nums[0];
    dp[1] = std::max(nums[0], nums[1]);
    
    // 填充 dp 数组
    for (int i = 2; i < n; ++i) {
        dp[i] = std::max(dp[i-1], dp[i-2] + nums[i]);
    }
    
    return dp[n-1];
}

int main() {
    std::vector<int> nums1 = {1, 2, 3, 1};
    std::cout << "输入: [1, 2, 3, 1]" << std::endl;
    std::cout << "输出: " << rob(nums1) << std::endl; // 输出: 4
    
    std::vector<int> nums2 = {2, 7, 9, 3, 1};
    std::cout << "输入: [2, 7, 9, 3, 1]" << std::endl;
    std::cout << "输出: " << rob(nums2) << std::endl; // 输出: 12
    
    return 0;
}

Java

public class HouseRobber {

    public int rob(int[] nums) {
        int n = nums.length;
        if (n == 0) return 0;
        if (n == 1) return nums[0];
        
        // 初始化 dp 数组
        int[] dp = new int[n];
        dp[0] = nums[0];
        dp[1] = Math.max(nums[0], nums[1]);
        
        // 填充 dp 数组
        for (int i = 2; i < n; ++i) {
            dp[i] = Math.max(dp[i-1], dp[i-2] + nums[i]);
        }
        
        return dp[n-1];
    }

    public static void main(String[] args) {
        HouseRobber robber = new HouseRobber();
        
        int[] nums1 = {1, 2, 3, 1};
        System.out.println("输入: [1, 2, 3, 1]");
        System.out.println("输出: " + robber.rob(nums1)); // 输出: 4
        
        int[] nums2 = {2, 7, 9, 3, 1};
        System.out.println("输入: [2, 7, 9, 3, 1]");
        System.out.println("输出: " + robber.rob(nums2)); // 输出: 12
    }
}

Python

def rob(nums):
    n = len(nums)
    if n == 0:
        return 0
    if n == 1:
        return nums[0]
    
    # 初始化 dp 数组
    dp = [0] * n
    dp[0] = nums[0]
    dp[1] = max(nums[0], nums[1])
    
    # 填充 dp 数组
    for i in range(2, n):
        dp[i] = max(dp[i-1], dp[i-2] + nums[i])
    
    return dp[n-1]

# 示例用法
nums1 = [1, 2, 3, 1]
print("输入:", nums1)
print("输出:", rob(nums1))  # 输出: 4

nums2 = [2, 7, 9, 3, 1]
print("输入:", nums2)
print("输出:", rob(nums2))  # 输出: 12

nums3 = [2, 1, 1, 2]
print("输入:", nums3)
print("输出:", rob(nums3))  # 输出: 4