139.最接近零的子数组和

描述

给定一个整数数组,找到一个和最接近于零的子数组。返回第一个和最有一个指数。你的代码应该返回满足要求的子数组的起始位置和结束位置

样例

给出[-3, 1, 1, -3, 5],返回[0, 2],[1, 3], [1, 1], [2, 2] 或者 [0, 4]

思路

求完和之后排个序,值接近的两个子数组和就相邻了

代码

class Pair {
    int sum;
    int index;
    public Pair(int s, int i) {
        sum = s;
        index = i;
    }
}
    
public class Solution {
    /**
     * @param nums: A list of integers
     * @return: A list of integers includes the index of the first number 
     *          and the index of the last number
     */
    public int[] subarraySumClosest(int[] nums) {
        int[] res = new int[2];
        if (nums == null || nums.length == 0) {
            return res;
        } 
        
        int len = nums.length;

        // 只有一个数时起始下标为0;
        if(len == 1) {
            res[0] = res[1] = 0;
            return res;
        }

        Pair[] sums = new Pair[len+1];
        // 前0个数之和
        // 这里是为只有第一个元素的子数组准备的 
        int prev = 0;
        // 在sums起始位置多添加一组数sums[0],方便计算
        sums[0] = new Pair(0, 0);
        // 给sum[]数组赋值
        // 无论是nums[]数组还是sums数组都是从下标0开始
        for (int i = 1; i <= len; i++) {
            // sums.sum表示前i个数之和,从nums开始数到第i-1个数
            sums[i] = new Pair(prev + nums[i-1], i);
            // prev后移一位
            prev = sums[i].sum;
        }

        // 自定义排序,sums是需要排序的数组,new Comparator<Pair>(){}是自定义的排序方式,按照sums.sum大小排序
        Arrays.sort(sums, new Comparator<Pair>() {
           public int compare(Pair a, Pair b) {
               return a.sum - b.sum;
           } 
        });

        int ans = Integer.MAX_VALUE;
        for (int i = 1; i <= len; i++) {
            // 排序后sums[i].sum - sums[i-1].sum一定是正值,满足if时更新ans
            if (ans > sums[i].sum - sums[i-1].sum) {
                // 寻找ans的最小值,即最接近于0的子数组和
                ans = sums[i].sum - sums[i-1].sum;
                // temp将得到的index排序,保证起始点的index不大于结束点的index
                int[] temp = new int[]{sums[i].index - 1, sums[i - 1].index - 1};
                // 调用默认的ASCII码排序
                Arrays.sort(temp);
                /* 因为sum的index是原数组中index为(0, i-1)的和,
                 * 存入temp的时候,index要减1才是真正的index;假如i > j ,
                 * 那么sum(0, i) - sum(0, j)表示的是(j + 1, i)这一段的和,所以要给小的那个加1
                 */
                res[0] = temp[0] + 1;
                res[1] = temp[1];
            }
        }
        
        return res;
    }
}

问:

为什么需要一个 (0,0) 的初始 Pair?

答:

我们首先需要回顾一下,在 subarray 这节课里,我们讲过一个重要的知识点,叫做 Prefix Sum
比如对于数组 [1,2,3,4],他的 Prefix Sum 是 [1,3,6,10]
分别表示 前1个数之和,前2个数之和,前3个数之和,前4个数之和
这个时候如果你想要知道 子数组 从下标 1 到下标 2 的这一段的和(2+3),就用前 3个数之和 减去 前1个数之和 = PrefixSum[2] - PrefixSum[0] = 6 - 1 = 5
你可以看到这里的 前 x 个数,和具体对应的下标之间,存在 +-1 的问题
第 x 个数的下标是 x - 1,反之 下标 x 是第 x + 1 个数
那么问题来了,如果要计算 下标从 0~2 这一段呢?也就是第1个数到第3个数,因为那样会访问到 PrefixSum[-1]
所以我们把 PrefixSum 整体往后面移动一位,把第0位空出来表示前0个数之和,也就是0. => [0,1,3,6,10]
那么此时就用 PrefixSum[3] - PrefixSum[0] ,这样计算就更方便了。
此时,PrefixSum[i] 代表 前i个数之和,也就是 下标区间在 0 ~ i-1 这一段的和
那么回过头来看看,为什么我们需要一个 (0,0) 的 pair 呢?
因为 这个 0,0 代表的就是前0个数之和为0
一个 n 个数的数组, 变成了 prefix Sum 数组之后,会多一个数出来

关于自定义Arrays.sort:

    Arrays.sort(objects, new Comparator<Object>() {  
            public int compare(Object object1, Object object2) {  
  
                    return object1.num - object2.num;  
            
        });  

比如这样,主要是重写compare方法,返回一个你需要的值来进行排序,这个返回值就是排序 的依据,比如要按object的num进行升序排序,就可以根据object1.num - object2.num来排序,如果返回的是object2.num - object1.num的话实际效果就是降序

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,015评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,262评论 1 292
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,727评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,986评论 0 205
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,363评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,610评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,871评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,582评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,297评论 1 242
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,551评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,053评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,385评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,035评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,079评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,841评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,648评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,550评论 2 270

推荐阅读更多精彩内容

  • 数组在程序设计中,为了处理方便, 把具有相同类型的若干变量按有序的形式组织起来。这些按序排列的同类数据元素的集合称...
    朱森阅读 3,780评论 2 13
  • 第1章 第一个C程序第2章 C语言基础第3章 变量和数据类型第4章 顺序结构程序设计第5章 条件结构程序设计第6章...
    小狮子365阅读 10,237评论 3 71
  • 感赏今天中午儿子冒着大太阳去市场帮我买来香菜,让我免去奔波的幸苦我知道外面很热,你在外面也打了一上午的球,本来就晒...
    爱娃林丽华aw阅读 187评论 0 2
  • 以前看到这些陌生的专业名词,会感觉好夸张,有点陌生,甚至怕怕的感觉。但经过这段时间的学习,不会在有这样的感觉...
    流年七里香农庄阅读 158评论 0 0
  • 现在的我,正坐在一个三本大学的自习室中,半个学期已经过去 ,从开始的好不情愿天天抱怨到现在的接受和进取,这是一个缓...
    龙侯侯阅读 142评论 0 0