zoukankan      html  css  js  c++  java
  • 和至少为K的最短子数组

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

    这道题的关键在于我们要知道各个区间的和。从而来判断哪个区间的和是满足要求的。用暴利解法逐个判断是可行的,但是耗费的时间太多。因为暴力解法其实做了很多重复的工作。那么为了节省重复的计算,有个方法就是保存数组的前缀和数组。也就是用一个数组sum来记录各个区间的和。也就是

    sum[i]=arr[0]+arr[1]+……+arr[i-1]。这里为了计算方便,增加了以为sum[0]=0。有了这个数组,就可以很方便的求解任意区间的和。

    比如要求区间[4,6]的和,就可以用sum[6]-sum[4]。

    那么接下来的工作就是要将sum数组中的各个求和值入队列。

    (1) 当区间[0,i]的值比[0,i-1]的值要大的时候,就直接进队列。

    (2) 当区间[0…i]的值比区间[0….i-1]还要小的时候,那么对于后面的位置,其实可以忽略掉i-1,因为sum[n]-sum[i] > sum[n]-sum[n-1]。也就是[i,n]区间的值比[i-1,n]的值要大,而且更短。因此i-1这个位置就可以被踢出队列。

    比如队列里面的数值如下:0,1,3,5,4。由于4比5小,此时应该将5剔除出队列,为什么呢,如果保留5在队列里面,当后面一个数组比如7进队列的时候,队列数组变成0,1,3,5,4,7。 这个时候计算区间和7-5明显比7-4要小。而且[5,7]的区间长度为2,[4,7]的区间长度为1。因此5没有必要存在于队列中。队里中的值变为0,1,3,4,7

    每次完成队列插入的操作后,就要开始对队列里面的区间进行计算了。

    当队列为0,1的时候,1-0<4 不满足条件

    当队列为0,1,3的时候,3-1<4 不满足条件

    当队列为0,1,3,5的时候,5-0>4 满足条件,最小长度为3。0出队列,队列变为1,3,5 继续比较5-1>4,最小长度为2,1出队列,队列变为3,5。

    当队列为3,4的时候(由于5比4小,因此出队列), 4-3<4不满足条件

    当队列为3,4,7的时候,7-3>=4,满足条件,最小长度为2

    代码如下:

    没有采用队列或者栈的结构,就用数组来存储dquue,用dq_begin, dq_end, dq_size来分别表示数组的第一个,最后一个元素以及数组长度。

    int shortestSubarray(int a[], int k, int len)

    {

        int *dqueue;

        int *sum;

        int i,dq_begin,dq_end,dq_size,min_length,lengh;

        dqueue = (int *)malloc((len + 1) * sizeof(int));

        sum = (int *)malloc(len * sizeof(int));

        memset(sum, 0,len * sizeof(int));

        memset(dqueue, 0, (len + 1) * sizeof(int));

        dq_begin = 0;

        dq_end = -1;

        dq_size = 0;

        min_length = len+1;

        for (i = 1; i <= len; i++)

        {

            *(sum + i) = *(sum+i-1)+a[i-1];

        }

        for (i = 0; i <= len; i++)

        {

            if (i != 0)

            {

                 while (dq_size > 0 && sum[dq_end] >= sum[i])

                 {

                     dq_end -= 1;

    dq_size -= 1;

                 }

                 while (dq_size > 0 && sum[i] - sum[dq_begin] >= k)

                 {

                     lengh = i - dq_begin;

                     min_length = min_length >= lengh ? lengh : min_length;

                     dq_begin += 1;

                     dq_size -= 1;

                 }

            }

            dq_end += 1;

            dqueue[dq_end] = i;

            dq_size += 1;

        }

        return min_length;

     

    }

  • 相关阅读:
    MySQL
    LeetCode
    数据结构
    我的编程幻想曲,更新中
    快速排序
    C与C++的区别
    必须要使用列表初始化的几种情况
    析构中delete this
    指向自身类型的成员指针的初始化,this不属于类对象的一部分
    构造函数
  • 原文地址:https://www.cnblogs.com/zhanghongfeng/p/11621746.html
Copyright © 2011-2022 走看看