一道简约而不简单的算法题——数据流的中位数

作者 | 程序员小吴

转载自微信公众号（ID:CXYxiaowu）

题目来源于 LeetCode 上第 295 号问题：数据流的中位数。难度级别为 Hard，目前通过率为 33.5% 。

题目描述

中位数是有序列表中间的数。如果列表长度是偶数，中位数则是中间两个数的平均值。

例如，

[2,3,4] 的中位数是 3

[2,3] 的中位数是 (2 + 3) / 2 = 2.5

设计一个支持以下两种操作的数据结构：

void addNum(int num) - 从数据流中添加一个整数到数据结构中。
double findMedian() - 返回目前所有元素的中位数。

示例：

addNum(1)addNum(2)findMedian() -> 1.5addNum(3) findMedian() -> 2

输入：

输出：

这道题给我们一个数据流，让我们找出中位数。对于数据流这种动态（流动）的数据，如果使用数组存储，那么每次新进来一个数据都进行排序的话，效率很低。

处理动态数据来说一般使用的数据结构是栈、队列、二叉树、堆。

本题中，我们使用堆这种数据结构。

首先将数据分为两部分，位于上边最大堆的数据要比下边最小堆的数据都要小。

为了保证将数据平均分配到两个堆中，在动态的操作的过程中两个堆中数据的数目之差不能超过 1。

为了保证最大堆中的所有数据都小于最小堆中的数据，在操作过程中，新添加进去的数据需要先和最大堆的最大值或者最小堆中的最小值进行比较。

动画描述

代码实现

class MedianFinder {    public PriorityQueue<Integer> minheap, maxheap;    public MedianFinder() {        //维护较大的元素的最小堆        maxheap = new PriorityQueue<Integer>(Collections.reverseOrder());        //维护较小元素的最大堆        minheap = new PriorityQueue<Integer>();    }

    // Adds a number into the data structure.    public void addNum(int num) {        maxheap.add(num);        minheap.add(maxheap.poll());        if (maxheap.size() < minheap.size()) {            maxheap.add(minheap.poll());        }    }

    // Returns the median of current data stream    public double findMedian() {        if (maxheap.size() == minheap.size()) {            return (maxheap.peek() + minheap.peek()) * 0.5;        } else {            return maxheap.peek();        }    }};

（本文为AI科技大本营转载文章，转载请联系原作者）

◆

实习生招募

◆

推荐阅读：

详解爱奇艺ZoomAI视频增强技术的应用 | 公开课笔记
DOTA2人机决战：2:0！OpenAI击败世界冠军OG
Python的10个“秘籍”，这些技术专家全都告诉你了
从头构建恶性肿瘤检测网络 | 100行Python代码理解深度学习关键概念
马云再谈 996：真正的 996 与被剥削无关
漫画：图的 “最短路径” 问题 | 技术头条
从 0 到管理 200 人，这位程序员是如何做到的？ | 程序员有话说
4000万假币流入波场, 发生在凌晨的BTT假币攻击事件始末及细节披露
程序员为什么都爱穿冲锋衣？（最全总结）

❤点击“阅读原文”，查看更多精彩文章。