双堆求中位数

来源:互联网 发布:大数据利用过程是 编辑:程序博客网 时间:2024/06/10 21:42

用堆维护中位数的意思是说,设计两个堆,第一个堆存放小于中位数的元素,第二个堆存放大于中位数的元素。

下面是双堆维护中位数的语言描述:

1、初始化的时候设置两个变量分别记录两个堆【左堆和右堆】的元素的个数

2、取第一个元素【d[0]】作为初始中位数m

3、循环后面的每一个元素,如果比m大,则插入到右堆,如果比m小,则插入到左堆

4、此时如果左堆和右堆的元素数目之差的绝对值等于2,说明左右两侧的元素不平衡了。需要把m插入到元素少的那个数组中,然后元素多的堆的堆顶元素作为中位数并将其从中删除,重复3和4。

5、经过上述步骤,数组的所有元素都已经在堆中了。如果两个堆的数据元素个数相等那么有两个中位数,即两个堆的堆顶。如果两个堆的数据数目不相等,则元素多的那个堆的堆顶是数组中位数。

原创粉丝点击