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
思路:用queue,Queue (Java Platform SE 7 ) , Queue is Interface. Implemented class : LinkedList, PriorityQueue.
class MovingAverage {
private Queue<Integer> queue;
private int size;
private double cursum;
public MovingAverage(int size) {
this.cursum = 0.0;
this.queue = new LinkedList<>();
this.size = size;
}
public double next(int val) {
if(queue.size() < size) {
queue.offer(val);
cursum += val;
} else {
Integer head = queue.poll();
cursum -= head;
queue.offer(val);
cursum += val;
}
return cursum / queue.size();
}
}
/**
* Your MovingAverage object will be instantiated and called as such:
* MovingAverage obj = new MovingAverage(size);
* double param_1 = obj.next(val);
*/