题目 给定一个整数数据流和一个窗口大小,根据该滑动窗口的大小,计算其所有整数的移动平均值。
示例:
MovingAverage m = new MovingAverage(3);
m.next(1) = 1
m.next(10) = (1 + 10) / 2
m.next(3) = (1 + 10 + 3) / 3
m.next(5) = (10 + 3 + 5) / 3
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/moving-average-from-data-stream
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class MovingAverage {
/** Initialize your data structure here. */
int size ;
List<Integer> list = new ArrayList();
public MovingAverage(int size)
{
this.size = size;
}
public double next(int val)
{
if(list.size()==size)
{
list.remove(0);
}
list.add(val);
double nums =0;
for(double x :list)
{
nums+=x;
}
nums /=list.size();
return nums;
}
}
定义成员变量 size,即滑动窗口的大小,具体底层采用数组实现,我们使用ArrayList,便于操作, 在next中检测窗口是否满了,如果满了,就剔除首个元素,并把新的元素加入list中,遍历求平均值即可,注意长度使用list的长度,而不是窗口大小,这样在窗口未满时也会得到正确的平均值