题目链接:长度最小的子数组
题目描述:
给定一个含有 n 个正整数的数组和一个正整数 s ,找出该数组中满足其和 ≥ s 的长度最小的连续子数组,并返回其长度。如果不存在符合条件的连续子数组,返回 0。
示例:
输入: s = 7, nums = [2,3,1,2,4,3]
输出: 2
解释: 子数组 [4,3] 是该条件下的长度最小的连续子数组。
题目分析:连续子数组和问题,两种思路方案:前缀和或者滑动窗口,下面分别贴出
前缀和:利用一个前缀数组来直接计算连续子数组的和,即sums[i]-sums[j]>=s,又题目中数组元素均为正数,那么前缀和是一个递增数组,我们已知s和某一个sums[i],可以利用二分查找出另一个sums[j],同时更新最小值。
代码:
class Solution {
public int minSubArrayLen(int s, int[] nums) {
int len = nums.length;
int l = 0,r = 0,sum = 0,ans = Integer.MAX_VALUE;
int[] sums = new int[len+1];
for(int i = 1;i <= len;i++)
sums[i] = sums[i-1] + nums[i-1];
for(int i = 0;i <= len;i++){
int te = s + sums[i];
int pos = Arrays.binarySearch(sums, te);
if(pos<0) pos = -pos-1;
if(pos<=len) ans = Math.min(ans,pos-i);
}
return ans == Integer.MAX_VALUE? 0: ans;
}
}
注意Arrays.binarySearch方法的返回值,找到就返回位置,找不到返回的是这个元素在原数组应该插入的位置的负数形式,因此需要处理一下。
滑动窗口:如果窗口中元素和小于s,则不作处理,如果大于等于s了,那么就将前面的元素不断删除,直到小于s,同时记录过程中窗口值的大小,其实理解为一个队列更加容易
以 s = 7, nums = [2,3,1,2,4,3] 为例
2
2,3
2,3,1
2,3,1,2 > 7 记录ans=4,并开始出队,2出->[3,1,2], < 7,停止出队、
3,1,2,4 > 7记录ans=4,开始出队,3出->[1,2,4] ,>=7 ,记录ans = 3,继续出队,1出->[2,4],<7停止
[2,4,3] > 7, 记录ans=3,出队,2出->[3,4] ,>= 7,记录ans = 2,继续出队,3出->[4],<7,停止
此时数组元素已经全部处理完,算法停止,ans = 2.
代码:
class Solution {
public int minSubArrayLen(int s, int[] nums) {
int len = nums.length;
int l = 0,r = 0,sum = 0,ans = Integer.MAX_VALUE;
while(r<len){
sum += nums[r++];//入队
while(sum>=s){//大于,开始出队
ans = Math.min(ans,r-l+1);//更新ans
sum -= nums[l++];//出队
}
}
return ans == Integer.MAX_VALUE? 0: ans;
}
}