Create main.cpp

Author: JawadSher

19 - Heap Data Structure Problems/16 - Find Median in a Stream/main.cpp

@@ -0,0 +1,29 @@
+class Solution
+{   
+    public:
+        priority_queue<int> maxHeap;
+        priority_queue<int, vector<int>, greater<int>> minHeap;
+        
+        void insertHeap(int &x)
+        {
+            if(maxHeap.size() == 0 || x < maxHeap.top()) maxHeap.push(x);
+            else minHeap.push(x);
+            
+            balanceHeaps();
+        }
+    
+        void balanceHeaps(){
+            if(maxHeap.size() > minHeap.size() + 1){
+                minHeap.push(maxHeap.top());
+                maxHeap.pop();
+            }else if(minHeap.size() > maxHeap.size()){
+                maxHeap.push(minHeap.top());
+                minHeap.pop();
+            }
+        }
+        
+        double getMedian(){
+            if(maxHeap.size() == minHeap.size()) return (minHeap.top() + maxHeap.top()) / 2.0; 
+            else return maxHeap.top();
+        }
+};