长度最小的子数组（Leetcode-209）-前缀和

题目

• 原题连接：https://leetcode-cn.com/problems/minimum-size-subarray-sum/
  

知识点

• 前缀和

思路

• 我们可以计算前缀和nums，然后再对前缀和数组按序遍历查找每一个num[i]对应的第一个（因为要求最小长度） 大于或等于nums[i] + target的元素nums[j]，这样得到j-i即是从i+1起满足条件第一个子数组的长度，我们每次对其更新最小长度。
• 由于全部都是正数，因此前缀和必然是递增的数组，所以在查找的时候可以使用二分查找的方式加快速度（c++：lower_bound()）。
• 在第一次发现不存在大于或等于nums[i] + target的元素，意味着后面的元素nums[i]都不可能找到大于或等于nums[i] + target的元素（因为是前缀和递增），直接break结束遍历即可。

代码

class Solution {
public:
    int minSubArrayLen(int target, vector<int> &nums) {
        int sum = 0;
        for (int &i:nums) {  // 计算前缀和
            int temp = i;
            i = sum;
            sum += temp;
        }
        nums.emplace_back(sum);  // 注意数组头插入一个0，因为第0个元素也有可能直接=target

        int min_len = INT32_MAX;
        for (int i = 0; i < nums.size() - 1; ++i) {
            int index = lower_bound(nums.begin(), nums.end(), nums[i] + target) - nums.begin();  //二分查找
            if (index < nums.size()){
                min_len = min(min_len, index - i);
            } else break;
        }
        if (min_len == INT32_MAX)   return 0;
        return min_len;
    }
};