forked from TheAlgorithms/Java
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request TheAlgorithms#581 from shrutisheoran/MedianOfRunni…
…ngArray_shrutisheoran Add Median Of Running Array
- Loading branch information
Showing
1 changed file
with
50 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,50 @@ | ||
import java.util.Collections; | ||
import java.util.PriorityQueue; | ||
|
||
/********************** | ||
author: shrutisheoran | ||
***********************/ | ||
|
||
public class MedianOfRunningArray { | ||
private PriorityQueue<Integer> p1; | ||
private PriorityQueue<Integer> p2; | ||
|
||
//Constructor | ||
public MedianOfRunningArray() { | ||
this.p1 = new PriorityQueue<>(Collections.reverseOrder()); //Max Heap | ||
this.p2 = new PriorityQueue<>(); //Min Heap | ||
} | ||
|
||
/* | ||
Inserting lower half of array to max Heap | ||
and upper half to min heap | ||
*/ | ||
public void insert(Integer e) { | ||
p2.add(e); | ||
if(p2.size() - p1.size() > 1) | ||
p1.add(p2.remove()); | ||
} | ||
|
||
/* | ||
Returns median at any given point | ||
*/ | ||
public Integer median() { | ||
if(p1.size()==p2.size()) | ||
return (p1.peek() + p2.peek())/2; | ||
return p1.size()>p2.size() ? p1.peek() : p2.peek(); | ||
} | ||
|
||
public static void main(String[] args) { | ||
/* | ||
Testing the median function | ||
*/ | ||
|
||
MedianOfRunningArray p = new MedianOfRunningArray(); | ||
int arr[] = {10, 7, 4, 9, 2, 3, 11, 17, 14}; | ||
for(int i = 0 ; i < 9 ; i++) { | ||
p.insert(arr[i]); | ||
System.out.print(p.median() + " "); | ||
} | ||
} | ||
|
||
} |