Amarkant814
diff --git a/‎July Leetcode Challenge 2021/11_medianDataStream.py
Copy file name to clipboard
+14Lines changed: 14 additions & 0 deletions b/‎July Leetcode Challenge 2021/11_medianDataStream.py
Copy file name to clipboard
+14Lines changed: 14 additions & 0 deletions
@@ -0,0 +1,14 @@
+from sortedcontainers import SortedList
+class MedianFinder:
+
+    def __init__(self):
+        self.arr = SortedList()
+
+    def addNum(self, num: int) -> None:
+        self.arr.add(num)
+
+    def findMedian(self) -> float:
+        n = len(self.arr)
+        if n % 2 == 1:
+            return self.arr[n//2]
+        return (self.arr[n//2] + self.arr[n//2-1]) / 2