forked from kdn251/interviews
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathMovingAverageFromDataStream.java
More file actions
37 lines (30 loc) · 979 Bytes
/
MovingAverageFromDataStream.java
File metadata and controls
37 lines (30 loc) · 979 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
// 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
/**
* Your MovingAverage object will be instantiated and called as such:
* MovingAverage obj = new MovingAverage(size);
* double param_1 = obj.next(val);
*/
public class MovingAverageFromDataStream {
double previousSum = 0.0;
int maxSize;
Queue<Integer> window;
/** Initialize your data structure here. */
public MovingAverage(int size) {
this.maxSize = size;
window = new LinkedList<Integer>();
}
public double next(int val) {
if(window.size() == maxSize) {
previousSum -= window.remove();
}
window.add(val);
previousSum += val;
return previousSum / window.size();
}
}