EASY/src/easy/MovingAveragefromDataStream.java

fishercoder1534 · fishercoder1534 · commit 9750986e75c6 · 2016-08-29T20:34:53.000-07:00
diff --git a/EASY/src/easy/MovingAveragefromDataStream.java b/EASY/src/easy/MovingAveragefromDataStream.java
@@ -0,0 +1,36 @@
+package easy;
+
+import java.util.LinkedList;
+import java.util.Queue;
+
+public class MovingAveragefromDataStream {
+
+}
+
+class MovingAverage {
+
+    private Queue<Integer> q;
+    private Long sum;
+    private int max;
+    
+    /** Initialize your data structure here. */
+    public MovingAverage(int size) {
+        q = new LinkedList();
+        sum = 0l;
+        max = size;
+    }
+    
+    public double next(int val) {
+        if(q.size() < max){
+            q.offer(val);
+            sum += val;
+            return (double) sum/q.size();
+        } else {
+            int first = q.poll();
+            sum -= first;
+            q.offer(val);
+            sum += val;
+            return (double) sum/q.size();
+        }
+    }
+}
