Minimum Size Subarray Sum 如果array中有negative number怎么办？

Given an array of n positive integers and a positive integer s, find the minimal length of a subarray of which the sum ≥ s. If there isn't one, return 0 instead.

For example, given the array [2,3,1,2,4,3] and s = 7,
the subarray [4,3] has the minimal length under the problem constraint.

我用的是 treemap 做的，做的时候没有用 positive number这个条件，可能是漏看了 = =

import java.util.Map.Entry;

public class Solution {
    /**
     * @param nums: an array of integers
     * @param s: an integer
     * @return: an integer representing the minimum size of subarray
     */
    public int minimumSize(int[] nums, int s) {
        if (nums == null || nums.length == 0) return -1;
        
        int minLen = Integer.MAX_VALUE;
        TreeMap<Long, Integer> map = new TreeMap<>();
        map.put(0L, -1);
        long sum = 0L;
        for (int i = 0; i < nums.length; ++i) {
            sum += nums[i];
            Entry<Long, Integer> floorEntry = map.floorEntry(sum - s);
            if (floorEntry != null && floorEntry.getKey() >= 0) {
                minLen = Math.min(minLen, i - floorEntry.getValue());
            }
            map.put(sum, i);
        }
        return minLen == Integer.MAX_VALUE ? -1 : minLen;
    }
}

思想就是用 sum 统计数组中加到第 i 个元素的和，然后从 treemap 里面找有没有之前加到某个元素(第 j 个)之和小于 sum - s，因为从j+1到 i之和肯定大于等于s

dp做？字数字数字数字数

celestial 发表于 2016-1-7 14:34
我用的是 treemap 做的，做的时候没有用 positive number这个条件，可能是漏看了 = =

import java.util. ...

感觉有bug，你作为对比的floorEntry的value（也就是index）不一定是所有满足条件的index里面最大的那个，这个在数组元素非负的时候确实成立，但是有负数的时候是不对的，考虑例子: array: [2,-1,-1,7], target: 5

你的code应该会输出3

163 发表于 2016-1-7 22:04
感觉有bug，你作为对比的floorEntry的value（也就是index）不一定是所有满足条件的index里面最大的那个， ...

You are right. Do you have any idea?

163 发表于 2016-1-7 22:04
感觉有bug，你作为对比的floorEntry的value（也就是index）不一定是所有满足条件的index里面最大的那个， ...

难道答案不就是3嘛？{-1,-1,7} ?

celestial 发表于 2016-1-11 11:34
You are right. Do you have any idea?

改成线段树就行了，复杂度也是O(nlogn)
（暂时想不到更简单的数据结构可以达到相同目的）

补充内容 (2018-6-26 18:51):
单调队列好像也行，代码短很多，但是复杂度仍然是O(nlogn)

神了，今天leetcode周赛最后一题就是这题，完全一模一样……

cai_lw 发表于 2018-7-1 10:15
神了，今天leetcode周赛最后一题就是这题，完全一模一样……

对的. 确实是这道.. 聪明的你!

这题我看了一眼答案, O[n]能做出来的. 其实想法也不能说很难, 确实很巧..

用的是deque, 只把比当前sum小的数字放在里面了..

肥宅快乐水 发表于 2018-7-4 04:20
对的. 确实是这道.. 聪明的你!

这题我看了一眼答案, O[n]能做出来的. 其实想法也不能说很难, ...

这个就是单调队列啊