動態規劃專題之----213. House Robber II

飄過的小熊發表於2017-04-22

原題連結

肯定是要在第一題的基礎來是比較好解決的。我走進了一個死衚衕是怎麼去解決首末同時在場的情況。而實際上是根本就不可能同時在場,既然不可能同時在場,那麼我就可以分別計算只有一個在場的情況下的值,通過比較得出解

  public int rob(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        int[] dp = new int[nums.length];
        dp[0] = nums[0];
        if (nums.length == 1) {
            return dp[0];
        }
        dp[1] = Math.max(nums[0], nums[1]);
        if (nums.length == 2) {
            return dp[1];
        }
        for (int i = 2; i < nums.length - 1; i++) {
            dp[i] = Math.max(dp[i - 1], dp[i - 2] + nums[i]);
        }
        int a = dp[nums.length - 2];//少最後一個值得情況


        int[] dp1 = new int[nums.length];
        dp1[1] = nums[1];
        dp1[2] = Math.max(nums[1], nums[2]);
        for (int i = 3; i < nums.length; i++) {
            dp1[i] = Math.max(dp1[i - 1], dp1[i - 2] + nums[i]);
        }
        int b = dp1[nums.length - 1];//少首部的情況

        if (a > b) {
            return a;
        } else {
            return b;
        }
    }

相關文章