leetcode-每日一题剑指 Offer II 041. 滑动窗口的平均值(队列模拟)

题目链接：https://leetcode.cn/problems/qIsx9U/

思路

方法一、队列模拟

直接想法

题目要求我们计算滑动窗口里所有数的平均值，给定了窗口大小size，我们在窗口里的数字个数不超过窗口大小时，按照个数计算平均值，一旦超过窗口大小，我们则需要移动窗口，计算当前窗口里的平均值

算法

1. 设计MovingAverage结构体存放窗口大小size，当前窗口数总和sum，记录当前窗口的数 num数组
2. Constructor 函数用于把窗口大小size记录在MovingAverage结构体里
3. Next方法用于计算窗口平均值
   1）当前窗口数字个数不超过窗口大小，按照个数计算平均值
   2）当前窗口数字个数超过窗口大小，需要移动窗口，计算当前窗口里的平均值

代码示例

type MovingAverage struct {
    size int
    sum int
    num []int
}

/** Initialize your data structure here. */
func Constructor(size int) MovingAverage {
    return MovingAverage{
        size: size,

    }
}

func (m *MovingAverage) Next(val int) float64 {
    //超过窗口大小，需要移动窗口
    if len(m.num) == m.size{
        m.sum -= m.num[0]
        m.num = m.num[1:]
    }
    m.sum += val
    m.num = append(m.num, val)
    return float64(m.sum) / float64(len(m.num))
}


/**
 * Your MovingAverage object will be instantiated and called as such:
 * obj := Constructor(size);
 * param_1 := obj.Next(val);
 */

复杂度分析

• 时间复杂度：O(1) 初始化和调用Next方法都只需要O(1)的时间
• 空间复杂度：O(size) 其中size是窗口大小，空间复杂度主要取决于窗口大小，窗口里的数字个数不会超过size