leetcode — 213 打家劫舍I II（动态规划入门）

题目一：

详解：

• 图解动态规划的解题四步骤
  

代码

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;
	}
}