Leetcode - 198 House Robber (Easy)
题目描述:一个小偷沿着一条街偷窃,给定数组表示每家屋子的金额,如果偷窃连续的两间屋子就会触发警报,求在不触发警报的前提下,偷窃的最大金额。
Input: [1,2,3,1]
Output: 4
Explanation: Rob house 1 (money = 1) and then rob house 3 (money = 3).
Total amount you can rob = 1 + 3 = 4.
解题思路:定义状态 dp[i] 表示从第 0 间到第 i 间屋子偷窃的最大金额,状态方程为:dp[i] = max(dp[i - 1], dp[i - 2] + num[i]),意为偷窃前 i 前的最大金额是「前 i - 1 间」与「前 i - 2 间加上第 i 间」两者的最大值。
public int rob(int[] num) {
if(num == null || num.length == 0) return 0;
int[] dp = new int[num.length + 1];
dp[0] = 0;
dp[1] = num[0];
for(int i = 1; i < dp.length - 1; i++) {
dp[i + 1] = Math.max(dp[i], dp[i - 1] + num[i]);
}
return dp[dp.length - 1];
}
上述代码中可以发现,计算过程中只需要两个变量即可完成计算。
public int rob(int[] num) {
if(num == null || num.length == 0) return 0;
int prev1 = 0, prev2 = 0;
for(int i = 0; i < num.length; i++){
int t = prev2;
prev2 = Math.max(prev1 + num[i], prev2);
prev1 = t;
}
return prev2;
}