数据流中的平均值
给定一个整数数据流和一个窗口大小,根据该滑动窗口的大小,计算其所有整数的移动平均值你的实现应该支持如下操作
示例:
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
没有限制,所以直接使用Queue和LinkedList即可。
import java.io.*;
import java.util.LinkedList;
import java.util.Queue;
class MovingAverage
{
private double sum = 0;
private int size;
private int num;
private Queue<Integer> que;
public MovingAverage(int size){
this.size = size;
que = new LinkedList<>();
}
public double next(int num){
this.num = num;
sum += num;
//判断长度
if(que.size()==size){
//减去头部
sum = sum - que.peek();
//移除头部
que.poll();
}
que.offer(num);
return sum/que.size();
}
public static void main (String[] args) throws java.lang.Exception
{
MovingAverage m = new MovingAverage(3);
System.out.println(m.next(1));
System.out.println(m.next(10));
System.out.println(m.next(3));
System.out.println(m.next(5));
}
}
基础题,单纯考察队列而已。