Leetcode编程C++之862. 和至少为 K 的最短子数组

Leetcode编程C++之862. 和至少为 K 的最短子数组

给你一个整数数组 nums 和一个整数 k ，找出 nums 中和至少为 k 的 最短非空子数组 ，并返回该子数组的长度。如果不存在这样的 子数组 ，返回 -1 。

子数组 是数组中 连续 的一部分

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/shortest-subarray-with-sum-at-least-k
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution {
public:
    static constexpr int INF = 0x3f3f3f3f;
    int shortestSubarray(const vector<int>& nums, int k) {
        const int n = nums.size();
        int ans = INF;
        long long s = 0;
        vector<pair<long long, int>> stk = {{0, -1}};
        for (int i = 0;i < n;++i) {
            s += nums[i];
            int l = 0, r = stk.size();
            while (l < r) {
                const int mid = (l + r) / 2;
                if (stk[mid].first + k <= s)
                    l = mid + 1;
                else r = mid;
            }
            if (l > 0) ans = min(ans, i - stk[l - 1].second);
            while (!stk.empty() && s <= stk.back().first)
                stk.pop_back();
            stk.emplace_back(s, i);
        }
        return ans < INF ? ans : -1;
    }
};