您的位置:首页 > Web前端

剑指offer_数据流中的中位数

2017-08-22 22:35 197 查看
/*
如何得到一个数据流中的中位数?
如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。
如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。

思路1:
采用ArrayList存储元素,每次添加一个元素就进行排序,排序调用Collections的sort函数,时间复杂度O(nlgn)
排序后的ArrayList可以在O(1)得到中位数

剑指offer思路:
1.不对数组进行排序,在O(1)时间插入元素,
采用partition在O(n)时间寻找中位数
2.每插入一个元素就进行排序,时间O(n),
在O(1)时间得到中位数
3.用链表存储元素,按顺序插入元素到链表,时间O(n),
用指针记录中间结点的位置,可以在O(1)时间得到中位数
4.用二叉搜索树,插入数据可以在平均O(lgn)时间完成,树极度不平衡时最差,O(n)
在二叉树中添加一个表示子树结点数目的字段,可以在平均O(lgn)时间找到中位数,怎么找?
5.利用平衡二叉树AVL,插入数据可以在O(lgn)时间完成,把AVL树的平衡因子改为左右子树结点数目差,可以在O(1)时间得到中位数

6.如果能保证数据容器左边的数据都小于右边的数据,这样即使两边内部的数据没有排序,也可以根据左边最大数和右边最小数得到中位数
左边使用最大堆,右边使用最小堆
为保证数据平均分配到两个堆中,数据总数目是偶数时新数据插入到最小堆中,否则插入到最大堆中
为保证左边的数据总数小于右边的数据,数据总数目是偶数时,先将新数据插入最大堆中,再将最大堆的最大元素取出插入最小堆
否则,先将新数据插入最小堆中,再将最小堆的最小元素取出插入最大堆
用什么来实现最大最小堆?要做到动态插入,参考算法导论优先队列

牛客网参考代码中采用最大最小堆的都直接调用优先队列PriorityQueue,自己如何实现?
*/

import java.util.*;

class  GetMedian
{
//思路1
ArrayList<Integer> al=new ArrayList<Integer>();
public  void insert(Integer num) {
al.add(num);
Collections.sort(al);
}

public  Double getMedian() {

int s=al.size();
double mid=0.0;
if (s%2==0)
{
mid=(double)(al.get(s/2-1)+al.get(s/2))/2;
}else
mid=al.get(s/2);

return mid;
}

//使用PriorityQueue

PriorityQueue<Integer> minHeap=new PriorityQueue<Integer>();

//PriorityQueue默认是小顶堆,实现大顶堆,需要反转默认排序器
PriorityQueue<Integer> maxHeap=new PriorityQueue<Integer>(20, new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o2.compareTo(o1);
}
});

int count=0;
public  void insert1(Integer num) {
count++;

if (count%2==0)
{
maxHeap.add(num);
int max=maxHeap.poll();
minHeap.add(max);
}else
{
minHeap.add(num);
int min=minHeap.poll();
maxHeap.add(min);
}
}

public  Double getMedian1() {

if (count%2==0)
{
int min=minHeap.peek();
int max=maxHeap.peek();
return (double)(min+max)/2;
}else

return maxHeap.peek();
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息