前言:双顶堆算法求是非常经典的一种求中位数算法,是堆必知必会的经典知识点。具体来说,就是如何求出数据流中的中位数。数据流的特点是高速插入,数据会不断涌入结构中,那么也就面临着需要多次动态调整以获得中位数。我们需要保证最大效率的情况下求出中位数,当然不能全部记录下来排序后求出中位数,这样每次插入都要排序一遍,非常消耗资源,当然最理想是用一个二叉搜索树来实现,但是面试的时候手撕二叉搜索树,难度颇大。本文介绍如何用一个大顶堆和一个小顶堆,实现求出数据流的中位数。
方法简介
我们用一个大顶堆+小顶堆的形式来求出中位数,控制所有小于等于中位数的数字放到一个堆中,控制所有比中位数大的数字放到另一个堆中,并且保证两个堆的数目相差小于等于1。这样就可以保证每一次查询中位数的时候,答案一定出于两个堆的堆顶元素之一。