📑 题目：198. 打家劫舍

🚀 本题 LeetCode 传送门

题目大意

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

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

解题思路

  • 你是一个专业的小偷，打算洗劫一条街的所有房子。每个房子里面有不同价值的宝物，但是如果你选择偷窃连续的 2 栋房子，就会触发警报系统，编程求出你最多可以偷窃价值多少的宝物？
  • 这一题可以用 DP 来解答，也可以用找规律的方法来解答。
  • DP 的状态定义是：dp[i] 代表抢 nums[0,i] 这个区间内房子的最大值，状态转移方程是 dp[i] = max(dp[i-1], nums[i]+dp[i-2]) 。可以优化迭代的过程，用两个临时变量来存储中间结果，以节约辅助空间。

代码

  1. package leetcode
  3. // 解法一 DP
  4. func rob198(nums []int) int {
  5.  n := len(nums)
  6.  if n == 0 {
  7.   return 0
  8.  }
  9.  if n == 1 {
  10.   return nums[0]
  11.  }
  12.  // dp[i] 代表抢 nums[0...i] 房子的最大价值
  13.  dp := make([]int, n)
  14.  dp[0], dp[1] = nums[0], max(nums[1], nums[0])
  15.  for i := 2; i < n; i++ {
  16.   dp[i] = max(dp[i-1], nums[i]+dp[i-2])
  17.  }
  18.  return dp[n-1]
  19. }
  21. // 解法二 DP 优化辅助空间，把迭代的值保存在 2 个变量中
  22. func rob198_1(nums []int) int {
  23.  n := len(nums)
  24.  if n == 0 {
  25.   return 0
  26.  }
  27.  curMax, preMax := 0, 0
  28.  for i := 0; i < n; i++ {
  29.   tmp := curMax
  30.   curMax = max(curMax, nums[i]+preMax)
  31.   preMax = tmp
  32.  }
  33.  return curMax
  34. }
  36. // 解法三 模拟
  37. func rob(nums []int) int {
  38.  // a 对于偶数位上的最大值的记录
  39.  // b 对于奇数位上的最大值的记录
  40.  a, b := 0, 0
  41.  for i := 0; i < len(nums); i++ {
  42.   if i%2 == 0 {
  43.    a = max(a+nums[i], b)
  44.   } else {
  45.    b = max(a, b+nums[i])
  46.   }
  47.  }
  48.  return max(a, b)
  49. }