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) / 3
思路:我们需要maintain一个sliding window, 每次都从头部加入一个元素进window,然后从尾部po出一个无效值,能完成这个操作的就是queue,因此我们选择queue来maintain这个window。代码如下:
int _size = 0;
int _index = 0;
double _sum = 0.0;
Queue<Integer> _queue = new LinkedList<>();
/** Initialize your data structure here. */
public MovingAverage(int size) {
_size = size;
}
public double next(int val) {
_sum += val;
_queue.offer(val);
if (_index++ >= _size) {
_sum -= _queue.remove();
}
//下面是我的代码,但是非常复杂冗余,最后发现其实可以用一句话实现,更加简洁美观,要懂得利用已有的信息!
// if (_index > _size) {
// double avr = _sum / (double)_size;
// return avr;
// } else {
// double avr = _sum / (double)_index;
// return avr;
// }
return _sum / _queue.size();
}