import java.util.PriorityQueue;
//如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,
// 那么中位数就是所有数值排序之后位于中间的数值。
// 如果从数据流中读出偶数个数值,那么中位数就是所有数值排序
// 之后中间两个数的平均值。我们使用Insert()方法读取数据流,
// 使用GetMedian()方法获取当前读取数据的中位数。
public class Test63 {
PriorityQueue<Integer> leftMaxHeap = new PriorityQueue<>(((o1, o2) -> o2.compareTo(o1)));
PriorityQueue<Integer> rightMinHeap = new PriorityQueue<>();
int count = 0;
public void Insert(Integer num) {
count++;
if ((count & 1) == 1) { //(count & 1) == 0为偶数,(count & 1) == 1为奇数
leftMaxHeap.add(num);
rightMinHeap.add(leftMaxHeap.poll());
} else {
rightMinHeap.add(num);
leftMaxHeap.add(rightMinHeap.poll());
}
}
public Double GetMedian() {
if ((count & 1) == 0) {
return (leftMaxHeap.peek() + rightMinHeap.peek()) / 2.0;
} else {
return (double) rightMinHeap.peek();
}
}
}
63 数据流中的中位数
最新推荐文章于 2023-01-18 21:53:57 发布