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.
For example, given the array [2,3,1,2,4,3] and s = 7,
the subarray [4,3] has the minimal length under the problem constraint.
算法思路
(1)以前曾经在一本ACM书上看到过尺度法,而这道题正是用这种方法解。
(2)这道题求得是在一个序列中,找到满足连续和为s的最短连续序列。
(3)当然,我们可以用dfs来做,但这样就没利用到“连续”这个信息,时间复杂度也会爆炸。
(4)其实题目很简单,想象自己有一把尺子,去量一个东西。短了?伸长一点。长了?缩短一点。
(5)我们建立两个指针指向连续序列的头和尾部,看下这个连续序列的和是多少。如果和大于s,那么就把头向后移动,相当于是缩短尺子,去掉了头部。如果小于s,那么尾指针向后移,相当于是伸长了尺子,增加了一个元素。我们只需要对比满足大于s的最小尺子长度即可。
算法代码
class Solution {
public:
int minSubArrayLen(int s, vector<int>& nums) {
int n = nums.size();
if(n == 0)
return 0;
int sum = nums[0];
int i = 0;
int j = 0;
int res = 99999;
if(sum >= s)
res = 1;
while(j < n){
if(sum < s){
j++;
if(j < n)
sum += nums[j];
}
else{
sum -= nums[i];
i++;
}
int tmp = j - i + 1;
if(sum >= s && res > tmp)
res = tmp;
}
if(res == 99999)
res = 0;
return res;
}
};