643. 子数组最大平均数 I
难度简单136收藏分享切换为英文接收动态反馈
给定 n
个整数,找出平均数最大且长度为 k
的连续子数组,并输出该最大平均数。
示例:
输入:[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]。
我的Java代码:
思路:题目求最大平均数,由于除数k是固定不变的,因此实际是求窗口内元素的最大和。用双指针维护动态窗口,计算出第一个窗口的和,当窗口移动时减去头指针元素加上尾指针元素即为当前窗口的和,并与历史最大值max比较。最终返回时用最大和除去k就可以了。
class Solution {
public double findMaxAverage(int[] nums, int k) {
int len = nums.length;
int head = 0, rear = k;
int sum = 0;
for(int i = head;i < rear;i++) {
sum += nums[i];
}
int max = sum;
while(rear <= len) {
if(rear+1 > len) {
break;
}
sum = sum-nums[head]+nums[rear];
if(sum > max) {
max = sum;
}
head++;
rear++;
}
return (double)max/(double)k;
}
}