算法通关村第十六关——

LeetCode643给定n个整数,找出平均数最大且长度为k的连续子数组,并输出该最大平均数。

其中1<=k<=nums.length<=105。

输入:[1,12,-5,-6,50,3],k = 4
输出:12.75
解释:最大平均数(12-5-6+56)/4 = 51/4 = 12.75
public static double findMaxAverage(int[] nums, int k) {
    int len = nums.length;
    int widowSum = 0;
    if (k > nums.length || nums.length < 1 || k < 1) {
        return 0;
    }
    // 第一步 先求第一个窗口的和
    for (int i = 0; i < k; i++) {
        widowSum += nums[i];
    }
    
    // 第二步 遍历,每次右边增加一个
    int res = widowSum;
    for (int right = k; right < len; right++) {
        widowSum += nums[right] - nums[right - k];
        res = Math.max(res, widowSum);
    }
    return (double) res / k;
}

widowSum += nums[right] - nums[right - k];这一步实现的就是向右移动一位,那么就需要加上右边的这位,减去最左边的那位。

在这里,很有一种在堆的学习那里的在数组中寻找第K大的元素,那时候的解决方法是先将前K个元素加入堆中,然后再从K开始依次遍历,这里也是一样的。

你可能感兴趣的:(算法,数据结构)