Week 13

Author: 8804who

find-median-from-data-stream/8804who.py

@@ -0,0 +1,26 @@
+import heapq
+
+class MedianFinder:
+
+    def __init__(self):
+        self.max_heap = []
+        self.min_heap = []
+
+    def addNum(self, num: int) -> None:
+        if not self.min_heap or self.min_heap[0] <= num:
+            heapq.heappush(self.min_heap, num)
+        else:
+            heapq.heappush(self.max_heap, -num)
+        
+        if len(self.min_heap) < len(self.max_heap):
+            heapq.heappush(self.min_heap, -heapq.heappop(self.max_heap))
+        elif len(self.min_heap) > len(self.max_heap) + 1:
+            heapq.heappush(self.max_heap, -heapq.heappop(self.min_heap))
+
+    def findMedian(self) -> float:
+        if len(self.min_heap) == len(self.max_heap):
+            return (self.min_heap[0]-self.max_heap[0])/2
+        else:
+            return self.min_heap[0]
+        
+