643.子数组最大平均数 I

子数组最大平均数 I

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

示例:
输入:[1,12,-5,-6,50,3], k = 4
输出:12.75
解释:最大平均数 (12-5-6+50)/4 = 51/4 = 12.75


解题思路

由于是关于一个数组中子数组的求和问题,所以可以利用滑动窗口方法来解。
(1) 首先,将数组 nums[] 中的前 k 个元素求和得到 sum ,并认为它是当前的最大值。
(2) 随后设置一个 left 指针用来指向子数组中的第一个元素。
(3) 遍历 nums[] 数组的后 n - k 个元素,每遍历一次就减去 nums[] 中 left 位置上的元素并加上最新进来的元素,计算完后与之前设置的最大值进行比较,直到遍历完所有元素。即可得到长度为k的连续子数组的最大值。


整个过程就像是有一个能够移动的窗口在这个数组 nums[] 上,从左至右一次移动一格。在计算时,每次都是计算当前这个窗口中的所有元素。

代码

class Solution {
   
   

    public double findMaxAverage(int[] nums, int k) {
   
   

        int sum = 0;
        int n = nums.length;
        for (int i = 0; i < k; i++) {
   
   

            //先把前k个元素全部相加
            sum += nums[i];
        }
        int maxSum = sum;
        int left = 0;//左指针--用来指向窗口中的第一个元素
        //从第k个开始到数组结尾(滑动窗口)---i就相当于是指向下一个要进入窗口的元素的指针
        for (int i = k; i < n; i++) {
   
   

            //nums[i]为新进入子数组的元素 nums[i-k]为原子数组中第一个元素
            sum = sum - nums[left++] + nums[i];
            maxSum = Math.max(maxSum, sum);
        }
        return 1.0 * maxSum / k;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章