题目
Given an array of n positive integers and a positive integer s, find the minimal length of a 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.
思路
我的思路是设置一个gap,从0开始增,看间隔为gap的区间和是否超过了s,超过则返回gap,没有则gap+1继续从头计算。这个复杂度是 n2 ,很明显不太好,代码也超时了,如下
int minSubArrayLen(int s, vector<int>& nums) {
int n = nums.size();
vector<vector<int>> sum(n,vector<int>(n,0));
for (int gap=0;gap<n;++gap){
for (int st=0;st+gap<n;++st){
if (gap==0) sum[st][st]=nums[st];
else sum[st][st+gap]=sum[st][st+gap-1]+nums[st+gap];
if (sum[st][st+gap]>=s) return gap+1;
}
}
return 0;
}
Hot解法
Hot解法给出了 O(n) 的解法,是设置两个下标,分别表示头和尾,首先移动尾,直到和超过sum,此时移动头,sum减去头指针指向的元素。当sum小于s的时候,记录头尾之间的间距,更新最小值。然后再重复移动尾指针,直至到头。这样做的好处是可以快速找到和超过s的一个区间,如果遍历一遍仍然不超过s则返回0即可,不需要像我的代码要 n2 的复杂度才跳出循环。
代码
int minSubArrayLen(int s, vector<int>& nums) {
int n = nums.size();
int start=0, end=0, sum=0, minlen=INT_MAX;
while(end<n){
while(end<n && sum<s) sum += nums[end++];
if (sum<s) break;
while (start<end && sum>=s) sum -= nums[start++];
if (end-start+1<minlen) minlen=end-start+1;
}
return minlen==INT_MAX?0:minlen;
}