описание темы
Как получить медиану в потоке данных? Если из потока данных считывается нечетное количество значений, медианой является значение в середине всех отсортированных значений. Если из потока данных считывается четное количество значений, медиана является средним значением двух средних чисел после сортировки всех значений.
Например,
медиана [2,3,4] равна 3.
Медиана [2,3] равна (2 + 3)/2 = 2,5.
Разработайте структуру данных, поддерживающую следующие две операции:
void addNum(int num) — добавляет целое число из потока данных в структуру данных.
double findMedian() — возвращает медиану всех элементов на данный момент.
ход мыслей
1. Сортировать список хранения, со вставкой и поиском.
2. Размер кучи - приоритетная очередь. При четных числах один человек кладет по половине большой и малой кучек, при нечетных числах к одной из двух кучек добавляется еще одна кучка.
фокус:
PriorityQueue<>((x, y) -> (y - x)) // 可以实现大根堆,原默认为小根堆。
Приведенный выше код также может использовать Comparator , то есть компаратор для настройки сортировки. Необходимо реализовать еще один метод сравнения.
PriorityQueue<Integer> maxHeap = new PriorityQueue<Integer>(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
/**以下是对比较器升序、降序的理解.
*(1) 写成return o1.compareTo(o2) 或者 return o1-o2表示升序
*(2) 写成return o2.compareTo(o1) 或者return o2-o1表示降序
*/
return o2.compareTo(o1);
}
}) ;
Методы операций в очереди — это добавление и опрос.
код
import java.util.*;
public class Solution {
Queue<Integer> max = new PriorityQueue<>(); //小堆
Queue<Integer> min = new PriorityQueue<>((x,y)->(y-x)); //大堆
//大小堆里插入
public void Insert(Integer num) {
if(max.size() != min.size()) {
max.add(num);
min.add(max.poll());
} else {
min.add(num);
max.add(min.poll());
}
}
//获取中位数
public Double GetMedian() {
double res = max.size() != min.size() ? max.peek() : (max.peek() + min.peek()) / 2.0;
return res;
}
}