Leetcode 209 - Minimum Size Subarray Sum（二分 or 双指针）

题意

给一个数组a[]和一个数s，要求在a里面找到一个长度最短的连续子序列，并且子序列的和大于等于s。

思路

算法1

前缀和 + 二分。时间复杂度$O(nlogn)$。

我们先维护一个前缀和ss，然后遍历ss里面的元素ss[i]，那么我们另一个需要的前缀和$ss_j$应该满足$ss_j \leq ss_i - s$。于是我们二分查找一下满足条件的$ss_j$即可。

算法2

双指针。时间复杂度$O(n)$。

我们用两个指针i和j，指向区间[i, j]代表当前的和sum的范围。

如果我们的$sum \geq s$，那么更新一下区间长度，并且可以将i后挪一位。

如果我们的$sum < s$，那么j后挪一位。

细节

vector.size()返回的是size_type，即unsigned_int，不能直接和负数做比较！必须强制转化为int才能做比较！可以用 typeid(x).name来查看x的类型。

代码

//algorithm1
class Solution {
public:
    int minSubArrayLen(int s, vector<int>& nums) {
        int *ss = new int[nums.size()];
        int ans = INT_MAX;
        for (int i = 0; i < nums.size(); i++) ss[i] = (i ? ss[i - 1] : 0) + nums[i];
        for (int i = 0; i < nums.size(); i++) {
            int x = ss[i] - s, k;
            if (x >= 0) {
                k = lower_bound(ss, ss + i, x) - ss;
                if (ss[k] > x) k--;
                ans = min(ans, i - k);
            }
        }
        return ans == INT_MAX ? 0 : ans;
    }
};

//algorithm2
class Solution {
public:
    int minSubArrayLen(int s, vector<int>& nums) {
        int ans = INT_MAX, i = 0, j = -1, sum = 0;
        while (i < nums.size() && j < (int)nums.size()) {
            if (sum >= s) {
                ans = min(ans, j - i + 1); 
                sum -= nums[i++];
            }
            else sum += nums[++j];
        }
        return ans == INT_MAX ? 0 : ans;
    }
};