leetcode 643 Maximum Average Subarray I

题目详情

Given an array consisting of n integers, find the contiguous subarray of given length k that has the maximum average value. And you need to output the maximum average value.

输入一个数组nums和一个整数k。要求找出输入数组中长度为k的子数组，而且要求子数组元素的加和平均值最大。返回这个最大的平均值。

Example 1:
Input: [1,12,-5,-6,50,3], k = 4
Output: 12.75
Explanation: 最大平均值 (12-5-6+50)/4 = 51/4 = 12.75

思路

• 创建一个长度为k的滑动窗口（即一个长度为k的子数组），而后每次右移一位，并将当前的平均值和存储的最大平均值比较，保留更大的那个值便可。

解法

public double findMaxAverage(int[] nums, int k) {
        double curr = 0;
        double max = 0;       
        
        for(int i=0;i<nums.length;i++){
            if(i < k){
                curr = curr + nums[i];
                max = curr;
                continue;
            }
            curr = curr + nums[i] - nums[i-k];    
            max = (curr > max) ? curr : max ;
            
        }
        
        
        return max/k;
        
    }