题目
返回 A 的最短的非空连续子数组的长度,该子数组的和至少为 K 。
如果没有和至少为 K 的非空子数组,返回 -1 。
示例 1:
输入:A = [1], K = 1
输出:1
示例 2:
输入:A = [1,2], K = 4
输出:-1
示例 3:
输入:A = [2,-1,2], K = 3
输出:3
提示:
1 <= A.length <= 50000
-10 ^ 5 <= A[i] <= 10 ^ 5
1 <= K <= 10 ^ 9
思路:这一题用普通的两个循环肯定是要超时的。可以选择用map将下表i
之前的数字的和s
保存起来,和s
作为key,i
作为value,当计算到s时,从map的key中选出满足s-key>=K
的Entry, 用当前的下标i 减去Entry的value,也就是满足和为key的下标(value),len = i - value
, 将len和minLength比较,minLength = min(len, minLength) ,更新minLength。
代码如下
class Solution {
public int shortestSubarray(int[] A, int K) {
int MAX = 50001;
TreeMap<Long, Integer> map = new TreeMap<Long, Integer>();
long sum = 0;
int minL = MAX;
long minSum = 100000000;
for (int i = 0; i < A.length; i++) {
sum += A[i];
if(sum >= K && minL > i + 1)
minL = i + 1;
long sub = sum - K;
while (!map.isEmpty()) {
Map.Entry<Long, Integer> subEntry = map.floorEntry(sub);
if (subEntry == null)
break;
long k = subEntry.getKey();
int j = subEntry.getValue();
if (i - j < minL) {
minL = i - j;
}
map.remove(k);
}
if (sum < minSum) {
minSum = sum;
map.clear();
}
map.put(sum, i);
}
if (minL == MAX)
return -1;
return minL;
}
}