LeetCode:862. 和最短为k的最短子数组[Java实现]

题目

返回 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;
    }
}