Leetcode:长度最小的子数组

题目描述

给定一个含有 n 个正整数的数组和一个正整数 s ，找出该数组中满足其和 ≥ s 的长度最小的连续子数组。如果不存在符合条件的连续子数组，返回 0。

示例 1:

输入: s = 7, nums = [2,3,1,2,4,3]
输出: 2
解释: 子数组 [4,3] 是该条件下的长度最小的连续子数组。

思路

可以使用滑动窗口的方法，i在左边界，j是右边界，j-i之间就是窗口大小，计算窗口之间元素和是否大于正整数 s,每次判断若sum小于s且还未到数组末尾，则sum加上窗口最右侧值，同时窗口右边界右移1，否则说明sum大于等于S或者已经到达数组末尾,此时sum减去左边界值，同时左边界右移1，要判断sum大于等于S时要更新最长子数组长度，最后判断一下若不存在符合条件的连续子数组，返回0

代码实现

class Solution {
public:
    //思路：可以使用滑动窗口的方法，i在左边界，j是右边界，j-i之间就是窗口大小，计算窗口之间元素和是否大于正整数 s
    //每次判断若sum小于s且还未到数组末尾，则sum加上窗口最右侧值，同时窗口右边界右移1，否则说明sum大于等于S或者已经到达数组末尾
    //此时sum减去左边界值，同时左边界右移1，要判断sum大于等于S时要更新最长子数组长度，最后判断一下若不存在符合条件的连续子数组，返回0
    int minSubArrayLen(int s, vector<int>& nums) {
        int length = nums.size();
        int i = 0, j = 0;
        int count = length + 1;//保证在后面比较时，可以取最小值，设置为最大值也行
        int sum = 0;
        while(i < length)
        {
            //若sum小于s且还未到数组末尾，则sum加上窗口最右侧值，同时窗口右边界右移1
            if(sum < s && j < length){
                sum += nums[j];
                ++j;
            }
            //否则说明sum大于等于S或者已到达数组末尾，此时sum减去左边界值，同时左边界右移1
            else{
                sum -= nums[i];
                ++i;
            }
            //判断sum大于等于S时要更新最长子数组长度
            if(sum >= s){
                count = count < (j - i) ? count : j-i;
            }
        }
        if(count == length + 1) {return 0;}
        return count;
    }
};