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

猜你喜欢

转载自blog.csdn.net/qq_36281031/article/details/108306197