Leetcode - House Robber II

Leetcode - House Robber II_第1张图片
Paste_Image.png

My code:

import java.util.Arrays;

public class Solution {
    public int rob(int[] nums) {
        if (nums == null || nums.length == 0)
            return 0;
        else if (nums.length == 1)
            return nums[0];
        return Math.max(rob(nums, 0, nums.length -1), rob(nums, 1, nums.length - 1));
    }
    
    private int rob(int[] nums, int begin, int length) {
        int[] dp = new int[length + 1];
        dp[0] = 0;
        dp[1] = nums[begin];
        for (int i = 2; i < length + 1; i++)
            dp[i] = Math.max(dp[i - 2] + nums[begin + i - 1], dp[i - 1]);
        Arrays.sort(dp);
        return dp[length];
    }
}

My test result:


Leetcode - House Robber II_第2张图片
Paste_Image.png

这道题目依然没有做出来。。。其实之前那道题目做出来后,这道题目应该要做出来。
但看到DP后,总是没有信心,于是再次没做出来。
我考虑的问题是,怎么把开头和结尾有没有抢辨别出来呢?记录下来呢?
这应该是大多数人思考的问题。
那么,有没有想过。假设房子标号0-n
他只有两种选择,要么从0开始抢,抢到n - 1
要么从1开始抢,抢到n。
当然,我说的从0开始抢,不代表就直接去抢0房子。而是代表0在他的考虑范围内。当抢到n-1后,即使n可以抢,他也不会去抢了。
然后再用我刚刚写的函数。
可以得到两个值,取较大的返回就行了。
还有个问题,如果数组只有一位,需要特别考虑下。

**
总结: DP
**

Anyway, Good luck, Richardo!

My code:

public class Solution {
    public int rob(int[] nums) {
        if (nums == null || nums.length == 0)
            return 0;
        else if (nums.length == 1)
            return nums[0];
        else if (nums.length == 2)
            return Math.max(nums[0], nums[1]);
        return Math.max(helper(nums, 0, nums.length - 2), helper(nums, 1, nums.length - 1));
    }
    
    private int helper(int[] nums, int begin, int end) {
        int[] dp = new int[end - begin + 1];
        dp[0] = nums[begin];
        dp[1] = Math.max(nums[begin], nums[begin + 1]);
        
        for (int i = begin + 2; i <= end; i++) {
            dp[i - begin] = Math.max(dp[i - begin - 2] + nums[i], dp[i - begin - 1]);
        }
        return dp[end - begin];
    }
}

这道题目,就是分成两种情况,
[0, n - 2]

[1, n - 1]

然后分别找最大。
返回较大者。

Anyway, Good luck, Richardo!

My code:

public class Solution {
    public int rob(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        else if (nums.length == 1) {
            return nums[0];
        }
        else if (nums.length == 2) {
            return Math.max(nums[0], nums[1]);
        }
        
        int n = nums.length;
        int[] dp1 = new int[n - 1]; // 0 - (n - 2)
        dp1[0] = nums[0];
        dp1[1] = Math.max(nums[0], nums[1]);
        for (int i = 2; i < dp1.length; i++) {
            dp1[i] = Math.max(nums[i] + dp1[i - 2], dp1[i - 1]);
        }
        
        int[] dp2 = new int[n - 1]; // 1 - (n - 1)
        dp2[0] = nums[1];
        dp2[1] = Math.max(nums[1], nums[2]);
        for (int i = 3; i < dp2.length + 1; i++) {
            dp2[i - 1] = Math.max(dp2[i - 3] + nums[i], dp2[i - 2]);
        }
        
        return Math.max(dp1[dp1.length - 1], dp2[dp2.length - 1]);
    }
}

还记得思路。没什么好说的。

Anyway, Good luck, Richardo! -- 08/25/2016

你可能感兴趣的:(Leetcode - House Robber II)