class Solution {
public int rob(int[] nums) {
int[][] dp = new int[nums.length][2];
if(nums.length == 1) return nums[0];
dp[0][0] = nums[0];
dp[0][1] = 0;
if(nums[1] > nums[0]){
dp[1][0] = nums[1];
dp[1][1] = 1;
} else {
dp[1][0] = nums[0];
dp[1][1] = 0;
}
for(int i = 2; i< nums.length;i++){
if(i == dp[i - 1][1] + 1){
if(dp[i - 2][0] + nums[i] > dp[i - 1][0]){
dp[i][0] = dp[i - 2][0] + nums[i];
dp[i][1] = i;
}else{
dp[i] = dp[i - 1];
}
}else{
dp[i][0] = dp[i - 1][0] + nums[i];
dp[i][1] = i;
}
}
return dp[nums.length - 1][0];
}
}
简单,当前状态往前推有三种情况分类讨论即可。
// 动态规划
class Solution {
public int rob(int[] nums) {
if (nums == null || nums.length == 0) return 0;
if (nums.length == 1) return nums[0];
int[] dp = new int[nums.length];
dp[0] = nums[0];
dp[1] = Math.max(dp[0], nums[1]);
for (int i = 2; i < nums.length; i++) {
dp[i] = Math.max(dp[i - 1], dp[i - 2] + nums[i]);
}
return dp[nums.length - 1];
}
}
好像不用考虑这么复杂,当i - 1的情况下,dp[i - 1][1] != i - 1,他的值等于dp[i - 2][0],所以是一样的。