346. Moving Average from Data Stream
Problem:
Use queue to keep the window.
Solution:
Given a stream of integers and a window size, calculate the moving average of all integers in the sliding window.
For example,
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) / 3Analysis:
Use queue to keep the window.
Solution:
class MovingAverage { int size; Queue<Integer> queue; double sum; /** Initialize your data structure here. */ public MovingAverage(int size) { this.size = size; queue = new LinkedList<>(); } public double next(int val) { queue.offer(val); sum += val; if (queue.size() > size) { sum -= queue.poll(); } return sum/queue.size(); } } /** * Your MovingAverage object will be instantiated and called as such: * MovingAverage obj = new MovingAverage(size); * double param_1 = obj.next(val); */
评论
发表评论