【Lintcode】617. Maximum Average Subarray II

题目地址：

https://www.lintcode.com/problem/maximum-average-subarray-ii/description

给定一个正整数$k$和一个长不少于$k$的整型数组$A$，求其平均数最大的且长度不少于$k$的子数组的平均值。

思路是二分答案。设$A$的最大值是$M$，最小值是$m$，那么答案一定在$[m,M]$里面。对于某个数$x$，我们要判断$A$是否有长度不少于$k$的且平均值大于等于$x$的子数组，如果有，则说明可以更新答案的左端点，否则更新右端点。至于判断这一点，可以将$A$的每个数都减去$x$，这样问题就转为是否存在长度不少于$k$的且和非负的子数组。这个可以通过维护最小前缀和，然后不停更新当前前缀和，看一下差是否非负。一旦发现非负，就说明存在。代码如下：

public class Solution {
    /**
     * @param nums: an array with positive and negative numbers
     * @param k:    an integer
     * @return: the maximum average
     */
    public double maxAverage(int[] nums, int k) {
        // write your code here
        // 求一下区间左右端点
        double l = nums[0], r = nums[0];
        for (int i = 1; i < nums.length; i++) {
            l = Math.min(l, nums[i]);
            r = Math.max(r, nums[i]);
        }
        
        double res = l;
        while (l + 1E-6 < r) {
            double m = l + (r - l) / 2.0;
            if (check(nums, k, m)) {
                l = m;
                res = Math.max(res, l);
            } else {
                r = m;
            }
        }
        
        return res;
    }
    
    private boolean check(int[] nums, int k, double avg) {
        double preL = 0, preR = 0, minPreL = 0;
        // 将长k的前缀赋值给preR
        for (int i = 0; i < k; i++) {
            preR += nums[i] - avg;
        }
        
        // 枚举区间右端点
        for (int i = k; i < nums.length; i++) {
            if (preR - minPreL >= 0) {
                return true;
            }
            
            preR += nums[i] - avg;
            preL += nums[i - k] - avg;
            minPreL = Math.min(minPreL, preL);
        }
        
        return preR >= minPreL;
    }
}

时间复杂度$O(l_A\log (M-m))$，空间$O(1)$。