# Median of Stream of Running Integers using STL

Given that integers are being read from a data stream. Find median of all the elements read so far starting from the first integer till the last integer. This is also called Median of Running Integers. The data stream can be any source of data, example: a file, an array of integers, input stream etc.

**What is Median?**

Median can be defined as the element in the data set which separates the higher half of the data sample from the lower half. In other words we can get the median element as, when the input size is odd, we take the middle element of sorted data. If the input size is even, we pick average of middle two elements in sorted stream.

**Example**:

Input: 5 10 15 Output: 5 7.5 10

**Explanation**: Given the input stream as an array of integers [5,10,15]. We will now read integers one by one and print the median correspondingly. So, after reading first element 5,median is 5. After reading 10,median is 7.5 After reading 15 ,median is 10.

The idea is to use max heap and min heap to store the elements of higher half and lower half. Max heap and min heap can be implemented using priority_queue in C++ STL. Below is the step by step algorithm to solve this problem.

**Algorithm**:

- Create two heaps. One max heap to maintain elements of lower half and one min heap to maintain elements of higher half at any point of time..
- Take initial value of median as 0.
- For every newly read element, insert it into either max heap or min heap and calulate the median based on the following conditions:
- If the size of max heap is greater than size of min heap and the element is less than previous median then pop the top element from max heap and insert into min heap and insert the new element to max heap else insert the new element to min heap. Calculate the new median as average of top of elements of both max and min heap.
- If the size of max heap is less than size of min heap and the element is greater than previous median then pop the top element from min heap and insert into max heap and insert the new element to min heap else insert the new element to max heap. Calculate the new median as average of top of elements of both max and min heap.
- If the size of both heaps are same. Then check if current is less than previous median or not. If the current element is less than previous median then insert it to max heap and new median will be equal to top element of max heap. If the current element is greater than previous median then insert it to min heap and new median will be equal to top element of min heap.

Below is C++ implementation of above approach:

`// C++ program to find med in ` `// stream of running integers ` `#include<bits/stdc++.h> ` `using` `namespace` `std; ` ` ` `// function to calculate med of stream ` `void` `printMedians(` `double` `arr[], ` `int` `n) ` `{ ` ` ` `// max heap to store the smaller half elements ` ` ` `priority_queue<` `double` `> s; ` ` ` ` ` `// min heap to store the greater half elements ` ` ` `priority_queue<` `double` `,vector<` `double` `>,greater<` `double` `> > g; ` ` ` ` ` `double` `med = arr[0]; ` ` ` `s.push(arr[0]); ` ` ` ` ` `cout << med << endl; ` ` ` ` ` `// reading elements of stream one by one ` ` ` `/* At any time we try to make heaps balanced and ` ` ` `their sizes differ by at-most 1. If heaps are ` ` ` `balanced,then we declare median as average of ` ` ` `min_heap_right.top() and max_heap_left.top() ` ` ` `If heaps are unbalanced,then median is defined ` ` ` `as the top element of heap of larger size */` ` ` `for` `(` `int` `i=1; i < n; i++) ` ` ` `{ ` ` ` `double` `x = arr[i]; ` ` ` ` ` `// case1(left side heap has more elements) ` ` ` `if` `(s.size() > g.size()) ` ` ` `{ ` ` ` `if` `(x < med) ` ` ` `{ ` ` ` `g.push(s.top()); ` ` ` `s.pop(); ` ` ` `s.push(x); ` ` ` `} ` ` ` `else` ` ` `g.push(x); ` ` ` ` ` `med = (s.top() + g.top())/2.0; ` ` ` `} ` ` ` ` ` `// case2(both heaps are balanced) ` ` ` `else` `if` `(s.size()==g.size()) ` ` ` `{ ` ` ` `if` `(x < med) ` ` ` `{ ` ` ` `s.push(x); ` ` ` `med = (` `double` `)s.top(); ` ` ` `} ` ` ` `else` ` ` `{ ` ` ` `g.push(x); ` ` ` `med = (` `double` `)g.top(); ` ` ` `} ` ` ` `} ` ` ` ` ` `// case3(right side heap has more elements) ` ` ` `else` ` ` `{ ` ` ` `if` `(x > med) ` ` ` `{ ` ` ` `s.push(g.top()); ` ` ` `g.pop(); ` ` ` `g.push(x); ` ` ` `} ` ` ` `else` ` ` `s.push(x); ` ` ` ` ` `med = (s.top() + g.top())/2.0; ` ` ` `} ` ` ` ` ` `cout << med << endl; ` ` ` `} ` `} ` ` ` `// Driver program to test above functions ` `int` `main() ` `{ ` ` ` `// stream of integers ` ` ` `double` `arr[] = {5, 15, 10, 20, 3}; ` ` ` `int` `n = ` `sizeof` `(arr)/` `sizeof` `(arr[0]); ` ` ` `printMedians(arr, n); ` ` ` `return` `0; ` `} ` |

*chevron_right*

*filter_none*

**Output:**

5 10 10 12.5 10

**Time Complexity:** O(n Log n)

**Auxiliary Space :** O(n)

This article is contributed by **Vibhu Garg**. If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to contribute@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.

Please write comments if you find anything incorrect, or you want to share more information about the topic discussed above.

## Recommended Posts:

- Median in a stream of integers (running integers)
- Median after K additional integers
- Minimum product of k integers in an array of positive Integers
- Mean and Median of a matrix
- Maximize the median of an array
- Median and Mode using Counting Sort
- Program for Mean and median of an unsorted array
- Find median of BST in O(n) time and O(1) space
- Median of two sorted arrays of different sizes
- Median of two sorted arrays of same size
- Randomized Algorithms | Set 3 (1/2 Approximate Median)
- Median of two sorted arrays with different sizes in O(log(min(n, m)))
- Find median in row wise sorted matrix
- Median of two sorted arrays of different sizes | Set 1 (Linear)
- Minimum number of elements to add to make median equals x