我一开始的想法是既然要找中位数,那肯定要排序,而且这个数据结构肯定要能动态的添加数据的,肯定不能用数组,于是我想到了用优先队列,它自己会排序都不用我写,所以addNum方法直接调用就可以,但是找中位数就很麻烦,它不能根据下标访问,于是我写了一个很屎的方法,把它poll到数组里面找到中位数后,再从数组放到优先队列中,代码如下:
class MedianFinder {
private PriorityQueue<Integer> priorityQueue;
/** initialize your data structure here. */
public MedianFinder() {
this.priorityQueue = new PriorityQueue<Integer>();
}
public void addNum(int num) {
priorityQueue.add(num);
}
public double findMedian() {
int size = priorityQueue.size();
int[] arr = new int[size];
for(int i=0;i<size;i++){
arr[i] = priorityQueue.poll();
}
for(int i=0;i<size;i++){
priorityQueue.add(arr[i]);
}
if(size%2 != 0){
return (double)arr[((size+1)/2) -1];
}else{
double res = (arr[(size/2)-1] + arr[size/2])/2.0;
return res;
}
}
}
但是它超时了,题目的测试用例非常大,我就想肯定是找中位数的时候太繁琐了,于是就想到了用LinkedList,可以通过下标获取中位数,只需要自己写一个排序,我就想到用add(int index, E element)方法,每次添加的时候找到添加的位置,这样就一直是有序的,复杂度是O(n),于是我又写了如下代码:
class MedianFinder {
private LinkedList<Integer> list;
/** initialize your data structure here. */
public MedianFinder() {
this.list = new LinkedList<Integer>();
}
public void addNum(int num) {
int size = list.size();
if(size == 0){
list.add(num);
return;
}else{
for(int i =0;i<size;i++){
if(i==0 && num <= list.get(0)){
list.addFirst(num);
break;
}else if(i==size-1 && num >= list.get(size-1)){
list.addLast(num);
break;
}else if(num >= list.get(i) && num <= list.get(i+1)){
list.add(i+1, num);
break;
}else{
continue;
}
}
}
}
public double findMedian() {
int size = list.size();
return size%2 !=0 ? list.get(((size+1)/2)-1)*1.0 : (list.get((size/2)-1)+list.get(size/2))/2.0;
}
}
但是tmd我没想到这也能超时,真进行了5000次的调用,然后我又试了一下用二分查找添加,不行,看题解了。题解看完我只想说一个字,妙!他也是用的优先队列,但是他用了两个优先队列,一个queMax存大于中位数的数,queMin存小于等于中位数的数,如果总数的奇数,那么中位数就是queMin的队头,如果总数是偶数那么中位数就是queMin和queMax的队头的平均值,添加的时候如果num小于等于中位数就加到queMin中,这时候新的中位数可能会小于原来的中位数,就要把queMin的对头放到queMax中,如果num大于中位数同理。
class MedianFinder {
private PriorityQueue<Integer> queMin;
private PriorityQueue<Integer> queMax;
/** initialize your data structure here. */
public MedianFinder() {
queMin = new PriorityQueue<Integer>((a, b) -> (b - a));
queMax = new PriorityQueue<Integer>((a, b) -> (a - b));
}
public void addNum(int num) {
if(queMin.isEmpty() || num <= queMin.peek()){
queMin.offer(num);
if(queMax.size() + 1 < queMin.size()){
queMax.offer(queMin.poll());
}
}else{
queMax.offer(num);
if(queMax.size() > queMin.size()){
queMin.offer(queMax.poll());
}
}
}
public double findMedian() {
if(queMin.size() > queMax.size()){
return queMin.peek();
}
return (queMax.peek() + queMin.peek()) / 2.0;
}
}
需要注意的是两个队列的排序方法不一样,queMin是队头最大,从大到小排,queMax是队头最小,从小到大排。所以看到两个优先队列的初始化方法是不一样的,lambda表达式中queMin的返回结果是b-a,而queMax是a-b。