【LeetCode】209. Minimum Size Subarray Sum

题目

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继续从头计算。这个复杂度是$n^2$，很明显不太好，代码也超时了，如下

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即可，不需要像我的代码要$n^2$ 的复杂度才跳出循环。

---

代码

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;
    }