La espada se refiere a la oferta 41. La mediana en el flujo de datos
Descripción del Título
Ideas para resolver problemas
Solución de la pregunta de referencia: pregunta 41 de la entrevista. Mediana en el flujo de datos (cola / montón de prioridad, ilustración clara)
La redacción al construir la gran pila superior:
maxHeap = new PriorityQueue<Integer>((x, y) -> (y - x));
Aquí (x, y) -> (y - x)están las nuevas características de Java lamda expresión 8 que se puede entender como una forma simplificada de la función de entrada, una función de los parámetros xey, la salida devuelve y - x.
Esta forma de escribir es equivalente a la siguiente:
maxHeap = new PriorityQueue<Integer>(new Comparator<Integer>(){
public int compare(Integer num1, Integer num2) {
return num2 - num1;
}
});
El código específico para esta pregunta es el siguiente:
class MedianFinder {
//小顶堆保存较大的元素，大顶堆保存较小的元素
private Queue<Integer> minHeap, maxHeap;
/** initialize your data structure here. */
public MedianFinder() {
minHeap = new PriorityQueue<Integer>();
maxHeap = new PriorityQueue<Integer>((x, y) -> (y - x));
}
public void addNum(int num) {
if (maxHeap.size() != minHeap.size()) {
//如果当前是奇数个元素，则需要往maxHeap中添加元素
minHeap.offer(num);
maxHeap.offer(minHeap.poll());
} else {
//如果当前是偶数个元素，则需要往minHeap中添加元素
maxHeap.offer(num);
minHeap.offer(maxHeap.poll());
}
}
public double findMedian() {
//如果是奇数，则返回minHeap的堆顶；如果是偶数，则返回两个堆堆顶的平均值
return maxHeap.size() != minHeap.size() ? (double)minHeap.peek() : (minHeap.peek() + maxHeap.peek()) / 2.0;
}
}
/**
* Your MedianFinder object will be instantiated and called as such:
* MedianFinder obj = new MedianFinder();
* obj.addNum(num);
* double param_2 = obj.findMedian();
*/