算法面试真题详解:子数组的最大平均值 II

简介: 算法面试真题详解:子数组的最大平均值 II

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

  • 保证数组的大小 >= k

在线评测地址:领扣题库官网

例1:

输入:
[1,12,-5,-6,50,3]
3
输出:
15.667
解释:
 (-6 + 50 + 3) / 3 = 15.667

例2:
输入:
[5]
1
输出:
5.000

###算法:二分答案
本题看到以后先想到暴力,即枚举所有可能子数组,时间复杂度O(N^2)会超时,于是我们考虑更低复杂度的做法,是否有时间复杂度O(NlogN)级别的做法呢?我们考虑二分答案来解决问题。
算法思路
- 我们考虑二分平均值,那么我们需要一个check函数,能在O(N)复杂度内判断是否存在一个子数组的平均值大于等于我们二分出来的平均值
- 对于一个平均数ave,我们先将nums数组每个数减去ave,那么只要存在一个长度大于k的子数组和大于等于0,就说明平均数ave可行,这可以在O(N)时间内完成
###代码思路
1. 设置二分的左右边界分别为数组中的最小值和最大值
2. 判断平均值mid是否可行,若可行则说明答案大于等于mid,那么左边界等于mid,否则说明答案小于mid,右边界等于mid
- 如何判断平均值mid是否可行:
  1. 将nums数组每个数减去mid
  2. 求nums数组的前缀和数组pre
  3. 设置指针index等于k
  4. 那么在nums[0:index]中,长度大于k的子数组,区间和最大为pre[index - 1] - min{pre[0 : index - k]}
  5. 将index不断右移直到指向数组末端,若中间区间和最大值大于等于0,check函数直接返回True,结束后还为返回值则返回False
3. 不断重复 2 直到 left + 1e-5 == right 退出
4. 返回左边界
###复杂度分析
NN表示nums数组长度,max_nums和min_nums分别表示数组中最大值和最小值
- 空间复杂度:O(1)
实际处理时不需要记录下整个前缀和数组,只需记录当前的前缀和和左侧最小的前缀和
- 时间复杂度:O(Nlog(max_nums−min_nums))

public class Solution {

/**
 * @param nums: an array with positive and negative numbers
 * @param k: an integer
 * @return: the maximum average
 */

private boolean check(int[] nums, int k, double avg) {

    //rightSum表示当前指向位置的前缀和
    //leftSum表示当前指向位置左侧k个位置的前缀和
    //minLeftSum表示左侧最小的前缀和

    double rightSum = 0, leftSum = 0, minLeftSum = 0;
    for (int i = 0; i < k; i++) {
        rightSum += nums[i] - avg;
    }
    for (int i = k; i <= nums.length; i++) {
        if (rightSum - minLeftSum >= 0) {
            return true;
        }
        if (i < nums.length) {
            rightSum += nums[i] - avg;
            leftSum += nums[i - k] - avg;
            minLeftSum = Math.min(minLeftSum, leftSum);
        }
    }
    return false;
} 

public double maxAverage(int[] nums, int k) {
    double left, right, mid;

    //设置二分的左右边界分别为数组中的最小值和最大值

    left = right = nums[0];
    for (int i = 0; i < nums.length; i++) {
        left = Math.min(nums[i], left);
        right = Math.max(nums[i], right);
    }
    while (left + 1e-5 < right) {
        mid = left + (right - left) / 2;

        //判断平均值mid是否可行
        //若可行则说明答案大于等于mid,那么左边界等于mid
        //否则说明答案小于mid,右边界等于mid

        if (check(nums, k, mid)) {
            left = mid;
        }
        else {
            right = mid;
        }
    }
    return left;
}

}

更多题解参考:[九章官网solution](https://www.jiuzhang.com/solution/maximum-average-subarray-ii/?utm_source=sc-tianchi-sz-20Dec)
相关文章
|
1月前
|
开发框架 算法 搜索推荐
C# .NET面试系列九:常见的算法
#### 1. 求质数 ```c# // 判断一个数是否为质数的方法 public static bool IsPrime(int number) { if (number < 2) { return false; } for (int i = 2; i <= Math.Sqrt(number); i++) { if (number % i == 0) { return false; } } return true; } class Progr
58 1
|
16天前
|
负载均衡 算法 应用服务中间件
面试题:Nginx有哪些负载均衡算法?Nginx位于七层网络结构中的哪一层?
字节跳动面试题:Nginx有哪些负载均衡算法?Nginx位于七层网络结构中的哪一层?
32 0
|
2月前
|
算法
|
1月前
|
算法
覃超老师 算法面试通关40讲
无论是阿里巴巴、腾讯、百度这些国内一线互联网企业,还是 Google、Facebook、Airbnb 等硅谷知名互联网公司,在招聘工程师的过程中,对算法和数据结构能力的考察都是重中之重。本课程以帮助求职者在短时间内掌握面试中最常见的算法与数据结构相关知识点,学会面试中高频算法题目的分析思路,同时给大家从面试官的角度来分析算法题的解答技巧,从而更有效地提升求职者的面试通过率。
15 3
覃超老师 算法面试通关40讲
|
1月前
|
存储 算法
【数据结构与算法】【腾讯阿里链表面试题】算法题--链表易懂版讲解
【数据结构与算法】【腾讯阿里链表面试题】算法题--链表易懂版讲解
|
1月前
|
存储 机器学习/深度学习 算法
python常用算法,新手必会,面试必出
python常用算法,新手必会,面试必出
37 0
|
1月前
|
存储 算法 Java
【数据结构与算法】2、链表(简单模拟 Java 中的 LinkedList 集合,反转链表面试题)
【数据结构与算法】2、链表(简单模拟 Java 中的 LinkedList 集合,反转链表面试题)
43 0
|
2月前
|
机器学习/深度学习 算法 Java
递归算法还有哪些是你不知道的----【探讨Java经典遍历问题和面试题】
递归算法还有哪些是你不知道的----【探讨Java经典遍历问题和面试题】
32 1
|
3月前
|
算法 测试技术 C#
【滑动窗口】C++算法:K 个不同整数的子数组
【滑动窗口】C++算法:K 个不同整数的子数组
|
28天前
|
Java 程序员
java线程池讲解面试
java线程池讲解面试
51 1