LeetCode643 子数组最大平均数 I

给定 n 个整数，找出平均数最大且长度为 k 的连续子数组，并输出该最大平均数。
示例：
输入：[1,12,-5,-6,50,3], k = 4
输出：12.75
解释：最大平均数 (12-5-6+50)/4 = 51/4 = 12.75
链接：https://leetcode-cn.com/problems/maximum-average-subarray-i/

目录

解题思路

代码

代码

---

解题思路

双指针算法
left 和 right 同时向右移动 , maxSum用来存储 left 和right 之间的最大值。
我自己写的算法时间复杂度为O(n^2) 提交LeetCode报超时。因为我每次移动都计算一遍left到right之间的和,这样是没有必要的。
可以将之前的计算的 sum 加上 nums[right+1] 然后减去nums[right-k]， 就可以得出移动之后left到right之间的和了。 而上一次的 sum 可以第一次就计算出来。
第二种参考答案的时间复杂度为O(n),又学到了呀。

代码

public static double findMaxAverage(int[] nums, int k) {
        if (nums.length < k) {
            int retSum = 0;
            for (int i : nums) {
                retSum += i;
            }
            return ((double) retSum) / k;
        }
        int maxSum = Integer.MIN_VALUE;
        for (int left = 0, right = k - 1; right < nums.length; left++, right++) {
            int sum = 0;
            for (int i = left; i <= right; i++) {
                sum += nums[i];
            }
            if (maxSum < sum) {
                maxSum = sum;
            }
        }
        return ((double) maxSum) / k;
    }

代码

    public static double findMaxAverage(int[] nums, int k) {
        int n = nums.length;
        int sum = 0;
        for (int i = 0; i < k; i++) {
            sum += nums[i];
        }
        int maxSum = sum;
        for (int i = k; i < n; i++) {
            sum = sum - nums[i - k] + nums[i];
            //可以不用记录 结果， 直接记录窗口和。返回时在计算平均值
            maxSum = Math.max(maxSum, sum);
        }
        return 1.0 * maxSum / k;
    }