题解 | #打家劫舍(二)#
打家劫舍(二)
https://www.nowcoder.com/practice/a5c127769dd74a63ada7bff37d9c5815
import java.util.*;
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param nums int整型一维数组
* @return int整型
*/
public int rob (int[] nums) {
// write code here
if(nums.length==1){
return nums[0];
}
int dp1[]=new int[nums.length+1];
int dp2[]=new int[nums.length+1];
first(nums,dp1);
twice(nums,dp2);
//找到最大值
int max=Math.max(dp1[nums.length-1],dp2[nums.length]);
return max;
}
//偷第一家,最后一家不偷
public void first(int[] nums,int dp1[]){
dp1[1]=nums[0];
for(int i=2;i<=nums.length;i++){
dp1[i]=Math.max(dp1[i-1],dp1[i-2]+nums[i-1]);
}
}
//偷第二家,最后一家偷
public void twice(int[] nums,int dp2[]){
dp2[1]=0;
for(int i=2;i<=nums.length;i++){
dp2[i]=Math.max(dp2[i-1],dp2[i-2]+nums[i-1]);
}
}
}
查看4道真题和解析