346-数据流中的移动平均值

题目描述

题目链接:346. 数据流中的移动平均值 - 力扣(LeetCode)

给定一个整数数据流和一个窗口大小,根据该滑动窗口的大小,计算其所有整数的移动平均值。

实现MovingAverage 类:

  • MovingAverage(int size) 用窗口大小 size 初始化对象。
  • double next(int val) 计算并返回数据流中最后 size 个值的移动平均值。

示例:

1
2
3
4
5
6
7
8
9
10
11
12
输入:
["MovingAverage", "next", "next", "next", "next"]
[[3], [1], [10], [3], [5]]
输出:
[null, 1.0, 5.5, 4.66667, 6.0]

解释:
MovingAverage movingAverage = new MovingAverage(3);
movingAverage.next(1); // 返回 1.0 = 1 / 1
movingAverage.next(10); // 返回 5.5 = (1 + 10) / 2
movingAverage.next(3); // 返回 4.66667 = (1 + 10 + 3) / 3
movingAverage.next(5); // 返回 6.0 = (10 + 3 + 5) / 3

提示:

  • 1 <= size <= 1000
  • $ -10^5 $<=val<= $ 10^5 $
  • 最多调用next方法 $10^4$ 次

基于数组的循环队列

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
class MovingAverage {
private int[] arr;
// 调用次数
private int tempSize;
// 窗口内数据和
private int sum;

/** Initialize your data structure here. */
public MovingAverage(int size) {
// 初始化
this.arr = new int[size];
}

public double next(int val) {
tempSize++;
sum += val;
// 计算游标
int index = (tempSize - 1) % arr.length;
// 将原数据从窗口中排除
if(tempSize > arr.length){
sum -= arr[index];
}
// 游标更新为新数据
arr[index] = val;
return (double) sum / (tempSize > arr.length ? arr.length : tempSize);

}
}

/**
* Your MovingAverage object will be instantiated and called as such:
* MovingAverage obj = new MovingAverage(size);
* double param_1 = obj.next(val);
*/

本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!