【算法】Shortest Subarray with Sum at Least K 求数组中区间和大于等于 K 的最小子数组长度

题目

Return the length of the shortest, non-empty, contiguous subarray of A with sum at least K.
If there is no non-empty subarray with sum at least K, return -1.

Example 1:
Input: A = [1], K = 1
Output: 1

Example 2:
Input: A = [1,2], K = 4
Output: -1

Example 3:
Input: A = [2,-1,2], K = 3
Output: 3

Note:
1 <= A.length <= 50000
-10 ^ 5 <= A[i] <= 10 ^ 5
1 <= K <= 10 ^ 9

找出数组 A 中其和大于等于 K 的最小、非空、连续子数组长度:

  1. A 中整数范围为 -10 ^ 5 <= A[i] <= 10 ^ 5
  2. A 数组长度范围为 1 <= A.length <= 50000
  3. K 的取值范围为 1 <= K <= 10 ^ 9

解题思路

本题求数组 A 中加和值 >=K 最小连续子数组长度,既然是连续子数组的问题,可以将数组 A 转换成累加数组 sums ,区间和[i, j] = sum[j] - sums[i-1],所以就转换成求 sums 中差值 >=K 的两个元素问题。
若 A 元素均为正数,以 A = [2, 1, 2] ,K = 2 为例:

  1. sums = [0, 2, 3, 5],为升序数组,不断将 sums[i] 与其之前的元素进行比较
  2. 二层遍历中,当找到 sums[1] - sums[0] >= K 满足条件
  3. 由于sums[i] (i>1) >= sums[1],肯定满足 >=K 的条件
  4. 但 i 距离 0 一定比 1 远,所以可以跳出循环,不用继续比较。

但 A 元素有负数的可能,sums 即非有序数组,所以无法在找到符合条件的两个元素后就终止二层遍历,但我们可以在二层遍历中直接过滤掉一些没必要的比较。

以 A = [1,1, -1, 2] ,K = 1 为例:

  1. sums = [0, 1, 2, 1, 3] ,为非有序数组
  2. 我们可以看到当 sums[4] 逐一对 sums[0 ... 3] 进行比较时,由于 sums[3] <= sums[2],则若 sums[4] - sums[2] >= K,那么 sums[4] - sums[3] >= sums[4]-sums[2] >= K 一定为真
  3. 而序号 4 - 3 < 4 - 2,所以 sums[4] 与 sums[2] 是没有必要比较的
  4. 同理 sums[4] 与 sums[1] 也是没有必要比较的
  5. 所以声明个比较队列 dq 来存储需要比较的序号
  6. 在遍历比较时,运用 sums[i] <= sums[j] 的方式,将 j 从比较队列中排除
  7. 同时在比较时更新最小值为结果

代码实现

func shortestSubarray(_ A: [Int], _ K: Int) -> Int {
        //A的长度n
        let n = A.count;
        //初始化结果res,初始值为max
        var res = Int.max;
        //初始化队列dq
        var dq = [Int]();
        //初始化累加队列sums
        var sums = [0];
        for a in A{
            sums.append(sums.last! + a)
        }
        //遍历sums
        for i in 0 ... n{
            //当队列不为空,且 [dq[0], i] 符合条件时,更新 res ,dq 删除头
            while (dq.count > 0 && sums[i] - sums[dq[0]] >= K){
                res = min(res, i - dq[0]);
                dq.remove(at: 0)
            }
            //当队列不为空,且 [dq.last, i] <= 0 时, dq 删除尾
            while (dq.count > 0 && sums[i] <= sums[dq.last!]){
                dq.remove(at: dq.count-1)
            }
            //将 i 加入 dq 尾部
            dq.append(i)
        }
        //返回结果
        return res == Int.max ? -1 : res
    }

代码地址:https://github.com/sinianshou/EGSwiftLearning

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

推荐阅读更多精彩内容