子数组最大平均数I
给定 n 个整数,找出平均数最大且长度为 k 的连续子数组,并输出该最大平均数。
示例 1:
输入: [1,12,-5,-6,50,3], k = 4
输出: 12.75
解释: 最大平均数 (12-5-6+50)/4 = 51/4 = 12.75
注意:
- 1 <= k <= n <= 30,000。
- 所给数据范围 [-10,000,10,000]。
思路
本题拿到手的时候,第一个思路就是对每一个元素都去求其长度为k的连续子数组的平均数,最后比较返回最大平均数,这样的算法时间复杂度~n*k,当k接近n时,时间复杂度~n平方,所以放弃这种算法。
我们初始化一个sum变量保存元素之和(初始化为0),一个maxAvg保存最大的平均数(初始化为Double下的负数最小值),这里需要强调一下, Double.NEGATIVE_INFINITY 表示负无穷,Double.MIN_VALUE 表示的是64位双精度值能表示的最小正数。然后开始遍历数组元素,每一步首先都将sum更新,然后判断当前索引是否大于等于k - 1,当等于k - 1时,对当前sum求平均,与maxAvg比较,当大于k - 1时,需要减去nums[i - k],这样就实现了动态更新sum,当小于k - 1时,继续循环。最后返回maxAvg即可。
1 public class Solution { 2 public double findMaxAverage(int[] nums, int k) { 3 /** 4 * Double.NEGATIVE_INFINITY 表示负无穷 5 * Double.MIN_VALUE 表示的是64位双精度值能表示的最小正数 6 */ 7 double maxAvg = Double.NEGATIVE_INFINITY; 8 double sum = 0.0; 9 for (int i = 0; i < nums.length; i++) { 10 sum += nums[i]; 11 if (i >= k - 1) { 12 if (i > k - 1) { 13 sum = sum - nums[i - k]; 14 } 15 if (sum / k > maxAvg) maxAvg = sum / k; 16 } 17 } 18 return maxAvg; 19 } 20 }
原文地址:https://www.cnblogs.com/kexinxin/p/10383057.html
时间: 2024-10-10 17:16:04