346. Moving Average from Data Stream

Given a stream of integers and a window size, calculate the moving average of all integers in the sliding window.
For example,
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

可以发现规律,数字长度不能超过定义的初始大小, 否则舍弃最开始的部分, 很明显要用到先进先出的队列。

public class MovingAverage { 
     Queue<Integer> queue;
     int queue_size;
     double sum = 0;
     public MovingAverage(int size) {
               queue = new LinkedList<>();
               queue_size = size;
     }
     public double next(int val) {
               sum += val;
               queue.offer(val);
               if(queue.size() > queue_size){
                       sum -= queue.poll();
                       return sum/queue_size;
               }else{
                       return sum/queue.size();
               }
     }
}

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容