中位数是有序列表中间的数。如果列表长度是偶数,中位数则是中间两个数的平均值。
例如,
[2,3,4] 的中位数是 3
[2,3] 的中位数是 (2 + 3) / 2 = 2.5
设计一个支持以下两种操作的数据结构:
void addNum(int num) - 从数据流中添加一个整数到数据结构中。 double findMedian() - 返回目前所有元素的中位数。 示例:
addNum(1) addNum(2) findMedian() -> 1.5 addNum(3) findMedian() -> 2 进阶:
如果数据流中所有整数都在 0 到 100 范围内,你将如何优化你的算法? 如果数据流中 99% 的整数都在 0 到 100 范围内,你将如何优化你的算法?
代码:暴力
class MedianFinder {
List<Integer> list=new ArrayList<>();
public MedianFinder() {
}
public void addNum(int num) {
list.add(num);
Collections.sort(list);
}
public double findMedian() {
int a=list.size();
if(a%2==0) {
return (list.get(a/2)+list.get(a/2-1))/2.0;
}
return list.get(a/2);
}
}
代码:
PriorityQueue<Integer> min;
PriorityQueue<Integer> max;
public MedianFinder() {
min=new PriorityQueue<>();
max=new PriorityQueue<>(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o2-o1;
}
});
}
public void addNum(int num) {
max.add(num);
min.add(max.remove());
if(min.size()>max.size()) {
max.add(min.remove());
}
}
public double findMedian() {
if(min.size()==max.size()) {
return (min.peek()+max.peek())/2.0;
}
return max.peek();
}
|