Сложность: easy
Дан поток целых чисел и размер окна, вычислите скользящее среднее всех целых чисел в скользящем окне.
Реализуйте класс MovingAverage:
MovingAverage(int size) Инициализирует объект с размером окна size.
double next(int val) Возвращает скользящее среднее последних size значений потока.
Пример:
Input
["MovingAverage", "next", "next", "next", "next"]
[[3], [1], [10], [3], [5]]
Output
[null, 1.0, 5.5, 4.66667, 6.0]
👨💻 Алгоритм:
1⃣Инициализация:
Инициализируйте объект с фиксированным размером окна size.
Используйте очередь или список для хранения значений в текущем окне.
Храните текущую сумму значений в окне.
2⃣Добавление нового значения:
Добавьте новое значение в очередь.
Обновите текущую сумму, добавив новое значение.
Если размер очереди превышает size, удалите самое старое значение и обновите сумму, вычтя это значение.
3⃣Вычисление скользящего среднего:
Возвращайте текущее скользящее среднее как сумму значений в окне, деленную на количество значений в окне.
😎 Решение:
class MovingAverage {
private var size: Int
private var queue: [Int]
private var sum: Int
init(_ size: Int) {
self.size = size
self.queue = []
self.sum = 0
}
func next(_ val: Int) -> Double {
queue.append(val)
sum += val
if queue.count > size {
sum -= queue.removeFirst()
}
return Double(sum) / Double(queue.count)
}
}Ставь 👍 и забирай 📚 Базу знаний