banner
cells

cells

为美好的世界献上 code

209. 长度最小的子数组

209. 长度最小的子数组#

给定一个含有 n 个正整数的数组和一个正整数 target

找出该数组中满足其总和大于等于 target 的长度最小的 连续子数组 [numsl,numsl+1,...,numsr1,numsr][nums_{l}, nums_{l+1}, ..., nums_{r-1}, nums_{r}],并返回其长度。如果不存在符合条件的子数组,返回 0

示例 1:

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

示例 2:

输入:target = 4, nums = [1,4,4]
输出:1

示例 3:

输入:target = 11, nums = [1,1,1,1,1,1,1,1]
输出:0

提示:

  • 1<=target<=1091 <= target <= 10^9

  • 1<=nums.length<=1051 <= nums.length <= 10^5

  • 1<=nums[i]<=1051 <= nums[i] <= 10^5

进阶:

  • 如果你已经实现 O(n) 时间复杂度的解法,请尝试设计一个 O(n log(n)) 时间复杂度的解法。

双指针#

class Solution {
public:
    int minSubArrayLen(int target, vector<int>& nums) {
        int n = nums.size();
        int i = 0, j = 0;
        int cur = 0;
        int res = INT_MAX;
        while (j < n) {
            cur += nums[j];
            while (cur >= target) {
                res = min(res, j - i + 1);
                cur -= nums[i];
                ++i;
            }
            ++j;
        }
        return res == INT_MAX ? 0 : res;
    }
};

前缀和#

区间和应联想到前缀和。

class Solution {
public:
    int minSubArrayLen(int target, vector<int>& nums) {
        int n = nums.size();
        vector<int> prev(n + 1, 0);
        // prev[i] 表示 nums 下标 [0, i) 的区间和
        for (int i = 1; i <= n; ++i) {
            prev[i] = prev[i - 1] + nums[i - 1];
        }
        int res = INT_MAX;
        for (int i = 0; i < n; ++i) {
            // prev[j] - prev[i] 表示 nums 下标 [i, j) 区间和
            // 需要满足 prev[j] - prev[i] >= target
            // 故 prev[j] >= target + prev[i]
            int find_val = target + prev[i];
            // prev 为非递减序列,可以使用二分法查找区间右坐标
            auto iter = lower_bound(prev.begin(), prev.end(), find_val);
            if (iter != prev.end()) {
                res = min(res, static_cast<int>(iter - prev.begin()) - i);
                // iter 在 prev 的下标比实际需要的多 1(j 为开区间),所以无需再加 1
            }
        }
        return res == INT_MAX ? 0 : res;
    }
};
加载中...
此文章数据所有权由区块链加密技术和智能合约保障仅归创作者所有。