最大平均值子数组

题目描述

给定n个数的数组,找到所有长度大于等于k的连续子数组中平均值最大的那个。返回那个最大的平均值。

1 <= k <= n <= 10000,数组中的元素在范围[-10000, 10000]之间,最后返回的答案的误差应在10^(-5)以内。

输入样例
输入: [1,12,-5,-6,50,3], k = 4
输出: 12.75

说明:
长度为4的子数组中,最大的平均值为12.75,(=(12 + -5 + -6 + 50)/4)

长度为5的子数组中,最大的平均值为10.8,(=(12 + -5 + -6 + 50 + 3)/5)

长度为6的子数组中,最大的平均值为9.16667。(所有数的平均值)

因此返回12.75。

解题思路

暴力枚举

枚举所有的长度大于等于k的子数组计算平均值,并从得到的平均值中求最大值,这样可以做到时间复杂度O(n^2),但是会超时。

二分法

有些最值问题可以转化为判断问题从而用二分法求得答案.
本题可以利用二分求解。 即
对于n个数a(0),a(1),……,a(n-1),以及一个数A,如果存在一个子数组起始于i,长为L>=k,使得其平均值大于等于A,即(a(i)+a(i+1)+……+a(i+L-1))/L >= A,那么我们所求的答案应当大于等于A;反之如果对于所有长度大于等于k的子数组,其平均值均小于A,那么我们所求的答案也必然小于A。

二分法的初始区间:
[min{a(i)},i=0~n-1 , max{a(i)},i=0~n-1]
因为在一组数中,该组数的平均值不会小于这组数的最小值,也不会大于这组数的最大值。同时,该组数的连续子数组的平均值 位于 这组数的最小值和最大值之间。

判断是否存在长度至少为k的子数组,其平均值大于等于A
式子(a(i)+a(i+1)+……+a(i+L-1))/L >= A,其等价于(a(i)-A)+(a(i+1)-A)+……+(a(i+L-1)-A)>=0,令b(0)=a(0)-A , b(1)=a(1)-A , …… , b(n-1)=a(n-1)-A,那么判断a数组中是否存在长度至少为k的子数组平均值大于等于A,等价于判断b数组中是否存在长度至少为k的子数组和大于等于0,只要求出b数组长度至少为k的子数组的最大和与0比较即可.

求长度大于等于k的最大和子数组:
求长度大于等于k的最大和子数组比原问题容易的多,令s为b的前缀和子数组,即s(i)=b(0)+b(1)+……+b(i-1),且s(0)=0,那么b(j)到b(i-1)的区间和可表示为s(i)-s(j),找长度大于等于k的最大和子数组等价于找i,j,满足i-j>=k,且使s(i)-s(j)最大。

固定i,则要使s(i)-s(j)最大,s(j)应最小,同时也应满足j<=i-k,令p(i) = min{s(j)},j<=i-k,故 固定 i 时s(i)-s(j)的最大值为s(i)-p(i),枚举所有i即可得到最终的最大值。因为s(i),p(i)均可通过递推得到,故时间复杂度为O(n)。

综上:总的时间复杂度: O(n * log(maxVal-minVal) / eps)

代码如下:

public double findMaxAvg(int[] A, int k){
        double left = Integer.MAX_VALUE;
        double right = Integer.MIN_VALUE;

        // 初始化 二分区间
        for(int i=0; i<A.length; ++i){
            left = Math.min(left, (double)A[i]);
            right = Math.max(right, (double)A[i]);
        }


        while(right-left > 1e-6){
            // 求出当前区间的中值
            double mid = (right-left)/2;

            // 求数组的累加和
            double[] sumAi = new double[A.length+1];
            sumAi[0] = 0;
            for(int i=0; i<A.length; i++){
                sumAi[i+1] = sumAi[i] + A[i] - mid;
            }

            // 求长度大于等于k的最大和子数组
            // 找长度大于等于k的最大和子数组等价于找i,j,满足i-j>=k,且使s(i)-s(j)最大
            // 最大化s(i),最小化s(j)
            double preMin = 0;
            double sumMax = Integer.MIN_VALUE;
            for(int i=k; i<=A.length; ++i){
                sumMax = Math.max(sumMax, sumAi[i] - preMin);
                preMin = Math.min(preMin, sumAi[i-k+1]);
            }

            // 判断是否存在长度大于等于k的字数组,其平均值大于等于 mid(sumMax>0)
            if(sumMax>0)
                left = mid;
            else right = mid;

        }
        return left;
    }

该算法的思路和实现基本参考了九章算法的题解。

斜率优化+单调队列解法

数列a中第 i 个位置到 第 j 个位置的平均值:avg(i, j) = (ai, ..., a) / (j-i+1).

若令sum(i) 表示数列a中从第 1 个位置 到 第 i 个位置 的累加和,则
avg(i, j) = (sum(j) - sum(i)) / (j - i + 1).

avg(i, j) 现在表示为 sum(.) 的斜率。

根据题意,我们的目标是找出子数组长度大于等于k,且求出子数组的最大平均值,即关于sum(.)的斜率最大的子数组。

假设我们已经得到 sum 数组。
由于数组的遍历,一般从左边开始, 因此我们假设 k < j < i. (i为要加入的点)
如果avg (k, j) > avg(j, i),表明 i 点 不应该被加入到解集中
,因为只有维护一个下凸的集合,我们才可以找到最大的斜率,如图1所示,只有我们维护了下凸集合,红色箭头会越来越陡峭,即斜率越来越大。

图1

由于屏蔽掉了要加入的点,相当于缩小了搜索空间。

在新加入点 i 后,如果 avg(y, x) < avg(x, i), 显然avg(y, i) < avg(x, i), y已经没有必要存在,因此可以从解集中去掉

图2

由于知道解集中的元素是单调递增的,我们可以使用单调队列保存解集。

// TODO code

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

推荐阅读更多精彩内容