题目一:
详解:
代码
package leetcode.dp;
public class Rob {
public int rob(int[] nums) {
int a = 0, b = 0; //a在前, b在后, 初始为nums[0]左边
for(int i=0;i<nums.length;i++) {
int cur = Math.max(a+nums[i], b);
a = b;
b = cur;
}
return b;
}
}
题目二:扩展版
- 环型街区
详解:
- 环状排列,即首尾相连,第一个房子和最后一个房子中只能偷一个,可以把环状排列房间问题化为两个单排街区子问题:
第一个子问题:不偷窃第一个房子的情况下(即 nums[1:]),最大金额是 p;
第二个子问题:不偷窃最后一个房子的情况下(即 nums[:n−1]),最大金额是 q。
然后取p和q的较大值。
代码
package leetcode.dp;
import java.util.Scanner;
public class Rob2 {
public static void main(String[] args) {
Scanner s = new Scanner(System.in);
while(s.hasNext()) {
int n = s.nextInt();
int[] nums = new int[n];
for(int i=0;i<n;i++) {
nums[i] = s.nextInt();
}
System.out.println(rob1(nums));
}
s.close();
}
public static int rob1(int[] nums) {
if(nums == null || nums.length == 0) return 0;
if(nums.length == 1) return nums[0];
int[] a = new int[nums.length-1]; //去掉第一家
int[] b = new int[nums.length-1]; //去掉最后一家
int j = 0;
int k = 0;
for(int i=0;i<nums.length;i++) {
if(i != 0)
a[j++] = nums[i];
if(i != nums.length-1)
b[k++] = nums[i];
}
int pre1 = 0, pre2 = 0;
for(int i=0;i<a.length;i++) {
int cur = Math.max(pre2, pre1 + a[i]);
pre1 = pre2;
pre2 = cur;
}
int max1 = pre2;
int pre3 = 0, pre4 = 0;
for(int i=0;i<b.length;i++) {
int cur = Math.max(pre4, pre3 + b[i]);
pre3 = pre4;
pre4 = cur;
}
int max2 = pre4;
return Math.max(max1, max2);
}
//精简写法:
public static int rob2(int[] nums) {
if(nums == null || nums.length == 0) return 0;
int len = nums.length;
if(len == 1) return nums[0];
return Math.max(robCore(nums, 0, len-2), robCore(nums, 1, len-1));
}
//不手动分割数组,传数组边界
public static int robCore(int[] nums, int start, int end) {
int pre1 = 0, pre2 = 0;
for(int i=start;i<=end;i++) {
int cur = Math.max(pre2, pre1 + nums[i]);
pre1 = pre2;
pre2 = cur;
}
return pre2;
}
}