关闭

【LeetCode】209. Minimum Size Subarray Sum

标签: leetcode
255人阅读 评论(0) 收藏 举报
分类:

题目

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

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:373328次
    • 积分:4860
    • 等级:
    • 排名:第5887名
    • 原创:162篇
    • 转载:18篇
    • 译文:0篇
    • 评论:70条
    博客专栏
    文章分类
    最新评论