1. 程式人生 > 其它 >字首和-長度最小的子陣列

字首和-長度最小的子陣列

題目連結:https://leetcode-cn.com/problems/minimum-size-subarray-sum
題目描述:
給定一個含有n個正整數的陣列和一個正整數 target 。
找出該陣列中滿足其和 ≥ target 的長度最小的 連續子陣列[numsl, numsl+1, ..., numsr-1, numsr] ,並返回其長度。如果不存在符合條件的子陣列,返回 0 。

示例 1:
輸入:target = 7, nums = [2,3,1,2,4,3]
輸出:2
解釋:子陣列[4,3]是該條件下的長度最小的子陣列。

示例 2:
輸入:target = 4, nums = [1,4,4]
輸出:1

示例 3:
輸入:target = 11, nums = [1,1,1,1,1,1,1,1]
輸出:0

提示:
1 <= target <= 109
1 <= nums.length <= 105
1 <= nums[i] <= 105

進階:
如果你已經實現 O(n) 時間複雜度的解法, 請嘗試設計一個 O(n log(n)) 時間複雜度的解法。

題解:

class Solution {
public:
    int minSubArrayLen(int s, vector<int>& nums) {
        
        int ans = nums.size() + 1;
        vector<int> prefix(nums.size() + 1, 0);    //字首和陣列
        prefix[0] = 0;
        for(int i =1; i < prefix.size(); i++)         
        {
            prefix[i] = prefix[i - 1] + nums[i - 1];
        }
        for(int i = 1; i < prefix.size(); i++)
        {
            int target = s + prefix[i - 1];    // prefix[j] - prefix[i] >= s
            int bound = binarySearch(target, prefix);
            if(bound < prefix.size() )
            {
                ans = min(ans, bound - (i - 1));
            }
        }
        return ans == nums.size() + 1 ? 0: ans;
    }
    int binarySearch(int t, vector<int>& prefix)   //查詢>=target的第一個數
    {
        int left = -1;
        int right = prefix.size();
        while(left + 1 != right)
        {
            int mid = (right + left) / 2;
            if(prefix[mid] < t)
            {
                left = mid;
            }else{
                right = mid;
            }

        }
        
        return right;
    }
};