这道题意思是,给出一个数组和一个目标和s,找一个最短的连续数组,它的和大于等于s,如果没有这样的数组,返回0。
首先想到的是遍历整个数组,求每个数后面不小于s的子数组的最短长度,找出最小的那个。
class Solution {
public:
int minSubArrayLen(int s, vector<int>& nums) {
int n = nums.size();
int res = INT_MAX;
for(int i = 0; i < n; ++i)
{
int sum = 0;
int start = i;
while(sum < s && start < n)
{
sum += nums[start++];
}
if(sum >= s)
res = min(res, start-i);
//cout << start << " " << i << endl;
}
if(res == INT_MAX) return 0;
return res;
}
};
另外有大神用二分搜索的方法:
我们需要定义两个指针left和right,分别记录子数组的左右的边界位置,然后我们让right向右移,直到子数组和大于等于给定值或者right达到数组末尾,此时我们更新最短距离,并且将left像右移一位,然后再sum中减去移去的值,然后重复上面的步骤,直到right到达末尾,且left到达临界位置,即要么到达边界,要么再往右移动,和就会小于给定值。
class Solution {
public:
int minSubArrayLen(int s, vector<int>& nums) {
if (nums.empty()) return 0;
int left = 0, right = 0, sum = 0, len = nums.size(), res = len + 1;
while (right < len) {
while (sum < s && right < len) {
sum += nums[right++];
}
while (sum >= s) {
res = min(res, right - left);
sum -= nums[left++];
}
}
return res == len + 1 ? 0 : res;
}
};