Leecode198.打家劫舍
题目
-
你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
-
给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
示例
-
输入:[1,2,3,1]
输出:4
解释:偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
偷窃到的最高金额 = 1 + 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
解题思路及算法详解
-
题目要求不能进入相邻的两个房间里,我们都是好孩子,就不说偷东西了,就类似打怪升级,每到一个房间,就能升相应的等级,这个题目就是典型的动态规划题目,最终获得最高的等级,
-
动态规划的三大步骤:
-
定义子问题
- 原问题要能由子问题表示。k=n 时实际上就是原问题。否则,解了半天子问题还是解不出原问题,那子问题岂不是白解了。一个子问题的解要能通过其他子问题的解求出。例如这道小偷问题中,f(k) 可以由 f(k-1) 和 f(k-2)) 求出,具体原理后面会解释。这个性质就是教科书中所说的“最优子结构”。如果定义不出这样的子问题,那么这道题实际上没法用动态规划解。
-
写出子问题的递推关系
-
f(k)=max{f(k−1),Hk−1+f(k−2)}
在写递推关系的时候,要注意写上 k=0 和 k=1 的基本情况:
当 k=0 时,没有房子,所以 f(0) = 0。
当 k=1 时,只有一个房子,偷这个房子即可,所以 f(1) = H0
-
-
确定 DP 数组的计算顺序
-
代码实现Java
-
public int rob(int[] nums) { if (nums.length == 0) { return 0; } // 子问题: // f(k) = 偷 [0..k) 房间中的最大金额 // f(0) = 0 // f(1) = nums[0] // f(k) = max{ rob(k-1), nums[k-1] + rob(k-2) } int N = nums.length; int[] dp = new int[N+1]; dp[0] = 0; dp[1] = nums[0]; for (int k = 2; k <= N; k++) { dp[k] = Math.max(dp[k-1], nums[k-1] + dp[k-2]); } return dp[N]; }
-
空间优化
-
public int rob(int[] nums) { int prev = 0; int curr = 0; for (int i : nums) { // 循环开始时,curr 表示 dp[k-1],prev 表示 dp[k-2] // dp[k] = max{ dp[k-1], dp[k-2] + i } int temp = Math.max(curr, prev + i); prev = curr; curr = temp; // 循环结束时,curr 表示 dp[k],prev 表示 dp[k-1] } return curr; }