LeetCode--198. House Robber & 213. House Robber II

题目链接:https://leetcode.com/problems/house-robber/和https://leetcode.com/problems/house-robber-ii/

问题一思路:很显然是一个动态规划问题,我们先用暴力搜索来做一遍,类似于Fibonacci数列的暴力求法,代码如下,是不是感觉跟Fibonacci数列求法基本一致呀:

public static int recursive(int idx,int[] nums)
	{
		if(idx==0)
			return nums[0];
		if(idx==1)
			return Math.max(nums[0], nums[1]);
		return Math.max(recursive(idx-2,nums)+nums[idx],recursive(idx-1,nums));
	}
	

	public static int rob(int[] nums) {
        if(nums.length==0)
			return 0;
		if(nums.length==1)
			return nums[0];
		return recursive(nums.length-1,nums);
    }

当然有了暴力求法和十分显然的递推公式,动态规划解法也不是很难想,设置一个当前最有解数组OPT来作为memo存储,这里的OPT[i]表示对于第i个屋子取或者不取两种决策获得的最优收益的较大值,也就是第i步决策后的最大收益。代码如下:

public static int rob(int[] nums) {
		if(nums.length==0)
			return 0;
		if(nums.length==1)
			return nums[0];
		int[] OPT=new int[nums.length+1];
		OPT[0]=nums[0];
		OPT[1]=Math.max(nums[0],nums[1]);
		
		for(int i=2;i<=nums.length;i++)
		{
			int value=(i==nums.length)?0:nums[i];				
			OPT[i]=Math.max(OPT[i-2]+value,OPT[i-1]);
		}
		return OPT[nums.length];
    }

问题二思路:因为最后一个屋子与第一个屋子相邻了,所以第一个取不取就是很关键,因为它涉及到最后一个屋子能不能取到,那就分两次动态规划来做,最后取两个DP最优解的较大值。值得注意的是,第一个屋子到底取不取,动态规划的初值是不一样的。举个栗子,帮大家理解一下:nums:[1,2,3,1]

nums[0]不取,OPT[0]=0,OPT[1]=nums[1]=2,OPT[2]=max(OPT[1],OPT[0]+nums[2])=3,OPT[3]=max(OPT[1]+nums[3],OPT[2])=3(最后一个取还是不取看情况,如果不取的解最大就不取,如果取的解最大则取),OPT[4]=max(OPT[3],OPT[2])=3。

nums[0]取,OPT[0]=nums[0],OPT[1]=OPT[0],OPT[2]=max(OPT[1],OPT[0]+nums[2])=4,OPT[3]=OPT[2](最后一个不取)

详细代码如下:

lass Solution {
    public int rob(int[] nums) {
        
        if(nums.length==0)
            return 0;
        if(nums.length==1)
            return nums[0];
        int[] skipFirstDP=new int[nums.length+1];
        skipFirstDP[0]=0;
        skipFirstDP[1]=nums[1];
        for(int i=2;i

 

你可能感兴趣的:(LeetCode--198. House Robber & 213. House Robber II)