膜拜笔者的引入思路:
讨论哪种数据结构存储数据 -> 引出数组 -> 排序的链表 -> 二叉搜索树 -> 优化的AVL树 -> 最后到堆,真是让人深感佩服。
// 以前写代码着重代码的简洁,现在写代码在复杂度相同的情况下,先可读性,再简洁性。
import java.util.PriorityQueue;
import java.util.Queue;
public class Solution {
Queue<Integer> minHeap = new PriorityQueue<>();
Queue<Integer> maxHeap = new PriorityQueue<>((o1, o2) -> o2 - o1);
int count = 0;
public void Insert(Integer num) {
if ((++count & 1) == 1) { // 数据总数为奇数放最大堆
if (!minHeap.isEmpty() && minHeap.peek() < num) {
int t = minHeap.poll();
minHeap.add(num);
maxHeap.add(t);
} else {
maxHeap.add(num);
}
} else { // 数据总数为偶数放最小堆
if (maxHeap.peek() > num) {
int t = maxHeap.poll();
maxHeap.add(num);
minHeap.add(t);
} else {
minHeap.add(num);
}
}
}
public Double GetMedian() {
if ((count & 1) == 1) {
return (double) maxHeap.peek();
} else {
return (maxHeap.peek() + minHeap.peek()) / 2.0;
}
}
}
京公网安备 11010502036488号