Add 346_Moving_Average_from_Data_Stream.java

seanprashad · Apr 15, 2022 · d3fb670 · d3fb670
1 parent 9d363d9
commit d3fb670
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/Design/346_Moving_Average_from_Data_Stream.java b/Design/346_Moving_Average_from_Data_Stream.java
@@ -0,0 +1,22 @@
+class MovingAverage {
+    private Queue<Integer> q;
+    private int sum;
+    private int maxSize;
+
+    public MovingAverage(int size) {
+        q = new LinkedList<>();
+        maxSize = size;
+        sum = 0;
+    }
+
+    public double next(int val) {
+        sum += val;
+        q.offer(val);
+
+        if (q.size() > maxSize) {
+            sum -= q.poll();
+        }
+
+        return (double) sum / q.size();
+    }
+}