862. 和至少为 K 的最短子数组

862. 和至少为 K 的最短子数组

返回 A 的最短的非空连续子数组的长度，该子数组的和至少为 K 。

如果没有和至少为 K 的非空子数组，返回 -1 。

示例 1：

输入：A = [1], K = 1
输出：1
示例 2：

输入：A = [1,2], K = 4
输出：-1
示例 3：

输入：A = [2,-1,2], K = 3
输出：3

提示：

1 <= A.length <= 50000
-10 ^ 5 <= A[i] <= 10 ^ 5
1 <= K <= 10 ^ 9

思路

用暴力解法肯定会超时的。这里采用了前缀和的方法，就是建立一个vector保存输入vector的前n项和；这样计算某一段和的时候，只要用前缀和对应两个位置的值相减就可以得到。再者，这里要找的时最短的数量，所以先暴力解start为0时的最短解，如果没有的话，就还是让num为n。之后的每一次搜索都控制从start开始的后num个和内，得到更小的num并更新它，所以这样搜索的次数会大大减少。不过呢，在leetcode中对于实例

在提交那时错的
输入：
[2,-1,2]
3
输出：
2
预期：
3
但是在playground里又时对的，让我不知道该做扫描操作了。

class Solution {
public:
    int shortestSubarray(vector<int>& A, int K) {
        int n = A.size(), cnt = A.size();
        vector<int> fsum(n+1, 0);
        for (int i = 1; i < n+1; i++) {
            fsum[i] = A[i-1] + fsum[i - 1];
        }
        for (int i = 0; i < n;i++) {
            for (int j = i+1; j-i < cnt&&j<n+1; j++) {
                //cout<<"i:"<<i<<"  "<<"j:"<<j<<"  "<<"fsum[i]:"<<fsum[i]<<"  "<<"fsum[j]:"<<fsum[j]<<endl;
                if (fsum[j]-fsum[i] >= K) {
                    cnt = j - i;
                    break;
                }
            }

        }
        if (cnt == n) { 
            if (accumulate(A.begin(), A.end(), 0) >= K)
                return cnt;
            else
                return -1;
        }
        return cnt;



    }
};