Solve : Find Median From Data Stream

Author: printjin-gmailcom

find-median-from-data-stream/printjin-gmailcom.py

@@ -0,0 +1,15 @@
+import heapq
+
+class MedianFinder:
+    def __init__(self):
+        self.low = []
+        self.high = [] 
+    def addNum(self, num):
+        heapq.heappush(self.low, -num)
+        heapq.heappush(self.high, -heapq.heappop(self.low))
+        if len(self.high) > len(self.low):
+            heapq.heappush(self.low, -heapq.heappop(self.high))
+    def findMedian(self):
+        if len(self.low) > len(self.high):
+            return -self.low[0]
+        return (-self.low[0] + self.high[0]) / 2