大厂面试真题详解：数据流滑动窗口平均值

给出一串整数流和窗口大小，计算滑动窗口中所有整数的平均值。

在线评测地址：领扣题库官网

样例1 :

MovingAverage m = new MovingAverage(3);
m.next(1) = 1 // 返回 1.00000
m.next(10) = (1 + 10) / 2 // 返回 5.50000
m.next(3) = (1 + 10 + 3) / 3 // 返回 4.66667
m.next(5) = (10 + 3 + 5) / 3 // 返回 6.00000

解题思路

我们需要一个数据结构来维护在窗口中的值，这个数据结构有一个要求，当存储的数个数大于窗口大小m时，需要弹出最先进入的数，这正好时队列的性质，所以可以用队列来维护。

算法：队列模拟滑动窗口

1. 初始化时新建一个队列。
2. 每次调用next(x)时，将x加入队列1。
3. 如果队列长度大于m，弹出队首。
4. 计算结果并返回。

复杂度分析

设窗口大小为m。
时间复杂度

• 单次next()时间复杂度O(1)
• 队列插入和弹出都是O(1)。
  空间复杂度
• 最多存储m个数，空间复杂度为O(m)。

public class MovingAverage {

    private Queue<Integer> queue;
    private double sum = 0;
    private int size;

    /** Initialize your data structure here. */
    public MovingAverage(int size) {
        queue = new LinkedList<Integer>();
        this.size = size;
    }
    
    public double next(int val) {
        sum += val;
        if (queue.size() == size) {
            sum = sum - queue.poll();
        }
        queue.offer(val);
        return sum / queue.size();
    }
}

更多题解参考：九章官网solution