数据结构学习(2) 数据流中的平均值

数据流中的平均值

给定一个整数数据流和一个窗口大小,根据该滑动窗口的大小,计算其所有整数的移动平均值你的实现应该支持如下操作

示例:

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 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));
	}
}

基础题,单纯考察队列而已。

你可能感兴趣的:(数据结构学习,java,队列)