一亩三分地论坛

 找回密码
 获取更多干货,去instant注册!

扫码关注一亩三分地公众号
查看: 1262|回复: 4
收起左侧

[算法题] Minimum Size Subarray Sum 如果array中有negative number怎么办?

[复制链接] |试试Instant~ |关注本帖
bcc 发表于 2015-12-22 07:25:21 | 显示全部楼层 |阅读模式

注册一亩三分地论坛,查看更多干货!

您需要 登录 才可以下载或查看,没有帐号?获取更多干货,去instant注册!

x
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.
celestial 发表于 2016-1-7 14:34:05 | 显示全部楼层
我用的是 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
回复 支持 1 反对 0

使用道具 举报

wxr.dal 发表于 2015-12-22 07:35:26 | 显示全部楼层
dp做?字数字数字数字数
回复 支持 反对

使用道具 举报

163 发表于 2016-1-7 22:04:36 | 显示全部楼层
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
回复 支持 反对

使用道具 举报

celestial 发表于 2016-1-11 11:34:27 | 显示全部楼层
163 发表于 2016-1-7 22:04
感觉有bug,你作为对比的floorEntry的value(也就是index)不一定是所有满足条件的index里面最大的那个, ...

You are right. Do you have any idea?
回复 支持 反对

使用道具 举报

本版积分规则

请点这里访问我们的新网站:一亩三分地Instant.

Instant搜索更强大,不扣积分,内容组织的更好更整洁!目前仍在beta版本,努力完善中!反馈请点这里

关闭

一亩三分地推荐上一条 /5 下一条

手机版|小黑屋|一亩三分地论坛声明 ( 沪ICP备11015994号 )

custom counter

GMT+8, 2016-12-9 18:02

Powered by Discuz! X3

© 2001-2013 Comsenz Inc. Design By HUXTeam

快速回复 返回顶部 返回列表