如何高效计算每日一题剑指Offer II 041题滑动窗口的平均值？

本文共计494个文字，预计阅读时间需要2分钟。

题目概述：https://leetcode.cn/problems/qIsx9U/实现一个`MovingAverage`类，使用窗口大小来计算滑动窗口内所有数字的平均值。

pythonclass MovingAverage: def __init__(self, size: int): self.size=size self.queue=[] self.sum=0

def add(self, val: int) -> float: if len(self.queue) >=self.size: self.sum -=self.queue.pop(0) self.queue.append(val) self.sum +=val return self.sum / len(self.queue)

题目见：leetcode.cn/problems/qIsx9U/

给定一个整数数据流和一个窗口大小，根据该滑动窗口的大小，计算滑动窗口里所有数字的平均值。

实现 MovingAverage 类：MovingAverage(int size) 用窗口大小 size 初始化对象。double next(int val) 成员函数 next 每次调用的时候都会往滑动窗口增加一个整数，请计算并返回数据流中最后 size 个值的移动平均值，即滑动窗口里所有数字的平均值。

解一：简单直接，用collections.deque 双端队列维持一个滑动窗口。

import collections class MovingAverage: def __init__(self, size: int): self.size = size self.queue = collections.deque() def next(self, val: int) -> float: self.queue.append(val) if len(self.queue) < self.size : return sum(self.queue) / len(self.queue) if len(self.queue) > self.size : self.queue.popleft() return sum(self.queue) / self.size

仔细看一下，其中if len(self.queue) &lt; self.size其实是没用的，可以优化掉。

import collections class MovingAverage: def __init__(self, size: int): self.size = size self.queue = collections.deque() def next(self, val: int) -> float: self.queue.append(val) if len(self.queue) > self.size : self.queue.popleft() return sum(self.queue) / len(self.queue)

解二：优化求和方式，用self.s 记录当前的和，每次滑动时只需在self.s上进行修改即可，不需要再对整个队列求和。

import collections class MovingAverage: def __init__(self, size: int): self.size = size self.s = 0 self.length = 0 self.queue = collections.deque() def next(self, val: int) -> float: self.queue.append(val) self.s += val self.length += 1 if self.length > self.size : t = self.queue.popleft() self.s -= t self.length -= 1 return self.s / self.length