上次错误:return dp[N]写成了return dp[N-1], 注意数组最后一个数 的索引是 数组长度-1,这里没注意数组长度。
上次错误:第一行特殊条件写错了,应该 if(nums.length==0)而不是if (nums == null)。
这道题是经典的动态规划入门问题。
动态规划四步骤:
- 定义子问题
- 写出子问题的递推关系
- 确定DP数组的计算顺序
- 空间优化(可选)
总结:首先,想到子问题的递推关系,想到递推关系的base case。然后,定义dp数组,也就是子问题数组。最后写出dp数组的base case和递推关系,在for循环中更新dp数组。
注意:因为前0项也占一个dp数组的位置,所以dp数组定义比原数组多,所以循环的结束条件也多1。
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 i=2;i<=N;i++) {
dp[i] = Math.max(dp[i-1], nums[i-1]+dp[i-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;
}