字首和-長度最小的子陣列
阿新 • • 發佈:2021-10-04
題目連結: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; } };