1. 程式人生 > >LeetCode—209—Minimum Size Subarray Sum(滑動視窗)

LeetCode—209—Minimum Size Subarray Sum(滑動視窗)

題目

Given an array of n positive integers and a positive integer s, find the minimal length of a contiguous subarray of which the sum ≥ s. If there isn’t one, return 0 instead.
Example:
Input: s = 7, nums = [2,3,1,2,4,3]
Output: 2
Explanation: the subarray [4,3] has the minimal length under the problem constraint.

翻譯

給定一個含有 n 個正整數的陣列和一個正整數 s ,找出該陣列中滿足其和 ≥ s 的長度最小的連續子陣列。返回這個最短的連續子陣列長度值,如果不存在符合條件的子陣列,返回 0。
輸入: s = 7, nums = [2,3,1,2,4,3]
輸出: 2
解釋: 子陣列 [4,3] 是該條件下的長度最小的子陣列。

解題思路

首先能想到的就是,找出所有的子陣列,並計算他們的和,然後判斷是否大於等於給定的目標值。並找到長度最小的那個子陣列。
用程式碼實現如下:

// 209. Minimum Size Subarray Sum
// https://leetcode.com/problems/minimum-size-subarray-sum/description/
// // 暴力解法 // 該方法在 Leetcode 中會超時! // 時間複雜度: O(n^3) // 空間複雜度: O(1) public class Solution1 { public int minSubArrayLen(int s, int[] nums) { if(s <= 0 || nums == null) throw new IllegalArgumentException("Illigal Arguments"); int res = nums.length + 1; for(int l = 0 ; l <
nums.length ; l ++) for(int r = l ; r < nums.length ; r ++){ int sum = 0; for(int i = l ; i <= r ; i ++) sum += nums[i]; if(sum >= s) res = Math.min(res, r - l + 1); } if(res == nums.length + 1) return 0; return res; } }

可以看出來這種解法,包含了三重迴圈。外面兩層找到所有子陣列,最內層計算當前陣列的sum值。時間複雜度為O(n^3),顯然不是一種優雅的解法。下面我們來看一種新的解法。俗稱滑動視窗。

優化解法

1.png

第一種解法之所以時間複雜度比較高呢,是包含了大量的重複運算。看上面的圖。如果我們知道了nums[i…j]的sum值,那麼nums[i…j-1]和nums[i…j+1]的sum值就特別容易得到。此時我們假設nums[i…j]連續陣列的和小於s。那麼此時我們將j指標向右移動一位;
2.png
移動到直到nums[i…j]的和大於或者等於s時。此時我們就找到了一個符合條件的連續子陣列,我們將他的長度記錄下來。
3.png
當前sum值大於或者等於s值。因為我們要求最短的子陣列。所以此時我們將左指標i向右移動。
2.png
直到nums[i…j]的sum值小於s。此時我們又繼續向右移動j。繼續找到一個連續子陣列使他的和是大於等於s的。依次類推。 那麼我們整個過程一直保持著一個視窗。這個視窗的長度和位置並不是固定不變的。是被i和j兩個索引決定的。這個視窗不斷向前滑動。來尋找滿足題意的連續子陣列。那麼這也就是滑動視窗的意思。下面我們來看程式碼實現。

// 209. Minimum Size Subarray Sum
// https://leetcode.com/problems/minimum-size-subarray-sum/description/
//
// 滑動視窗的思路
// 時間複雜度: O(n)
// 空間複雜度: O(1)
public class Solution3 {

    public int minSubArrayLen(int s, int[] nums) {

        if(s <= 0 || nums == null)
            throw new IllegalArgumentException("Illigal Arguments");
        // nums[l...r]為我們的滑動視窗
        //因為這個區間是閉區間。初始情況視窗是沒有元素的。所以r的值為-1
        int l = 0 , r = -1;
        int sum = 0;
        //這裡res是代表最小陣列的長度
        //將他初始值設為整個陣列長度加一。方便了比較以及判斷最終是否找到解。
        int res = nums.length + 1;

        while(l < nums.length){   // 視窗的左邊界在陣列範圍內,則迴圈繼續

            if(r + 1 < nums.length && sum < s)  //判斷r+1是保證陣列不越界
                sum += nums[++r];
            else // r已經到頭 或者 sum >= s
                sum -= nums[l++];

            if(sum >= s)
                res = Math.min(res, r - l + 1);  //因為是閉區間,所以陣列的長度為r-l+1
        }

        if(res == nums.length + 1)
            return 0;
        return res;
    }
}

關注我免費下載CSDN

關注公眾號哦