Create find_median_from_stream.cpp

Author: avidLearnerInProgress

find_median_from_stream.cpp

@@ -0,0 +1,25 @@
+class MedianFinder {
+public:
+    priority_queue<int> maxHeap;
+    priority_queue<int, vector<int>, greater<> > minHeap;
+    /*
+        1) store first half in maxheap, second half in minheap
+        2) maxheap can have 1 element more than minheap
+    */
+    void addNum(int num) {
+        maxHeap.push(num);
+        minHeap.push(maxHeap.top());
+        maxHeap.pop();
+        
+        if(minHeap.size() > maxHeap.size()){
+            maxHeap.push(minHeap.top());
+            minHeap.pop();
+        }
+    }
+    double findMedian() {
+        if(maxHeap.size() == minHeap.size()) {
+            return (maxHeap.top() + minHeap.top()) / 2.0;
+        }
+        return maxHeap.top();
+    }
+};