862. 和至少为 K 的最短子数组
给你一个整数数组 nums 和一个整数 k ,找出 nums 中和至少为 k 的 最短非空子数组 ,并返回该子数组的长度。如果不存在这样的 子数组 ,返回 -1 。
子数组 是数组中 连续 的一部分。
本题关键点为和至少、连续,看到这些我们可以前缀和加队列进行解题。需要注意的是前缀和此处使用long类型,如果不留意结果就不尽人意。
代码如下:
public int shortestSubarray(int[] nums, int k) {
// 获取前缀和
long[] array = new long[nums.length+1];
for(int i=0;i< nums.length;i++){
array[i+1] = array[i] + nums[i];
if(nums[i]>=k){
return 1;
}
}
int ans = Integer.MAX_VALUE;
Deque<Integer> deque = new ArrayDeque<>();
for(int i=0;i < array.length;i++){
// 如果当前前缀和比队首的差大于等于K 说明找到一个符合条件子序列且最短(将左侧弹出)
while(!deque.isEmpty() && (array[i] - array[deque.getFirst()]) >= k){
ans = Math.min(ans,i-deque.getFirst());
deque.removeFirst();
}
// 此时队列中剩下的都是符合条件的数据 就开始求和最优解(和最小且大于K)
// 如果前缀和当前比队尾的要小 那么将队尾去掉
while(!deque.isEmpty() && array[i] <= array[deque.getLast()]){
deque.removeLast();
}
// 将当前加入队列右侧,因为当前前缀和更小更优
deque.addLast(i);
}
return ans == Integer.MAX_VALUE ? -1 : ans;
}
运行结果如下:
(含有注释效果会低,去掉注释后内存由提升)
个人见解不喜勿喷!!!