Java 算法-最大平均值子数组 (二分法)

  今天遇到了一道题,本来题的不难的,但是就是因为加入了很多的限制,导致了这个题比较棘手。

题意

给出一个整数数组,有正有负。找到这样一个子数组,他的长度大于等于 k,且
平均值最大。

样例

给出 nums = [1, 12, -5, -6, 50, 3], k = 3

返回 15.667 // (-6 + 50 + 3) / 3 = 15.667

注意事项

保证数组的大小 >= k

1.传统的方法(超时)

  这个题的传统非常的容易,这里不便再赘述

public static double maxAverage(int[] nums, int k) {
        //记录当前k(或者大于k)位数字和
        double sum = 0;
        double max = Integer.MIN_VALUE;
        if(nums == null || nums.length == 0 || k == 0){
            return 0;
        }
        for(int j = k; j <= nums.length; j++){
            //初始化为0
            sum = 0;
            for(int i = 0; i < nums.length; i++){
                if(i >= j){//当i > j时,计算最大值
                    max = ((sum * 1.0 / j) > max) ? sum * 1.0 / j : max;
                    sum -= nums[i - j]; //需要减去当前选择的数字的第一个,因为即将在加入一个
                }
                //加入一个数字
                sum += nums[i];
            }
            max = ((sum * 1.0 / j) > max) ? sum * 1.0 / j : max;
        }
        return max;
    }

2.二分法

  对的,又是二分法。跟前面的快慢指针一样,这里也是非常规的二分法使用。

(1).题意介绍

  这个题让我们求的是一个数组的子数组(长度大于等于k)的最大平均值,记住不是最大值,是最大平均值。因为得出最大值不一定能够推出最大的平均值,因为平均值不仅受和的影响,还要受个数影响。

(2).算法概述

  首先一个数组的和的平均值一定在这个数组中的最小值和最大值之间。这一点肯定没有争议。
  其次,我们现在要做的就是--二分,在这个范围里面,不断的二分,直到找到我们想要的值,但是又应该怎么二分呢?
  我们可以这样来假设,假设 最大值为max,最小值为min,中间值为 mid = (min + max ) /2.0(这里为什么要除以2.0,而不是2呢?那是我们在[min, max]区间求得我们想要的那个平均值,而这个平均值不可能总是整数,有可能是小数(double),所以,我们除以2.0而不是2)。假设nums数组(nums[0], nums[1], nums[2], nums[3].......),如果存在两个下标i,j(i > j),长度i - j >= k,使其(相当于是nums数组的子数组)平均值大于等于mid的值,那么:

(nums[i] + nums[i + 1] + ...... + nums[j]) / (i - j) >= mid.

  所以,得出我们想要的答案必定在[mid, max]区间里面;反之,如果不存在这两个下标,也就是所有的子数组的平均值都小于mid,那么我们求得最终的答案肯定小于mid,所以最终的答案肯定在[min, mid]。
  那么,怎么判断一个子数组(长度L >= k)和的平均值大于mid呢?
  这个是我们这个问题的一个难点,我们用数学公式来表达,如图所示:



  如图上的结论,我们还可以假设s数组,其中

                      s[0] = 0;

                      s[1] = b[0];

                      s[2] = b[0] + b[1] = s[1] + b[1];

                      s[3] = b[0] + b[1] + b[2] = s[2] + b[2];

   那么b[j]到b[i](i > j)区间和为s[i + 1] - s[j]。要想找出区间和(长度大于等于k)的最大值,等价于找i,j,满足i - j >= k并且使得s[i] - s[j]最大。要想求出s[i + 1] - s[j]的最大值,我们固定i,只要s[j]最小的话,那么s[i + 1] - s[j]就最大了。

(3).代码

public static double maxAverage(int[] nums, int k) {
        if (nums == null || nums.length == 0 || k == 0) {
            return 0;
        }
        double max = Integer.MIN_VALUE;
        double min = Integer.MAX_VALUE;
        for(int i = 0; i < nums.length; i++){
            if(nums[i] > max){
                max = nums[i];
            }
            if(nums[i] < min){
                min = nums[i];
            }
        }
        while (max - min >= 1e-6) {
            double mid = (max + min) / 2.0;
            if(binarySearch(nums, mid, k)){//表示最大的平均值在[min, mid]里面
                min = mid;
            }else{//表示最大的平均值在[mid, max]里面
                max = mid;
            }
        }
        return min;
    }

    private static boolean binarySearch(int[] nums, double mid, int k) {
        //用来记录区间和的
        double sum[] = new  double[nums.length + 1];
        double min = 0;
        sum[0] = 0;
        for (int i = 1; i <= nums.length; i++) {
            //计算每个点的区间和
            sum[i] = sum[i - 1] + (nums[i - 1] - mid);
            //min相当于是s[j],不过这里是一步一步的取小值得出来的
            if (i >= k && sum[i] >= min) {//表示有b数组的和>=0,则取值范围变为[mid, max]
                return true;
            }  if (i >= k) {
                min = Math.min(min, sum[i - k + 1]); //一步一步的取小值
            }
        }
        return false;
    }
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,287评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,346评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,277评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,132评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,147评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,106评论 1 295
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,019评论 3 417
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,862评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,301评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,521评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,682评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,405评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,996评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,651评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,803评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,674评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,563评论 2 352

推荐阅读更多精彩内容