LeetCode 295 Find Median from Data Stream

最新推荐文章于 2021-01-12 15:56:52 发布

EricJeffrey

最新推荐文章于 2021-01-12 15:56:52 发布

阅读量177

点赞数

分类专栏： leetcode 算法文章标签：数据结构

本文链接：https://blog.csdn.net/q15037911903/article/details/112303824

版权

算法同时被 2 个专栏收录

40 篇文章 0 订阅

订阅专栏

leetcode

11 篇文章 0 订阅

订阅专栏

该博客介绍了一种使用multiset数据结构来实现实时中位数查找的算法。通过维护一个multiset并更新指向中位数的指针，可以在数据插入后快速找到中位数。代码中定义了一个MedianFinder类，其addNum方法用于添加数值，并根据条件调整指针位置，findMedian方法则返回当前数据集的中位数。这种方法保证了在数据插入后的迭代器稳定性。

摘要由CSDN通过智能技术生成

思路

使用multiset+pointer，在向multiset插入数据同时更新pointer，确保pointer总是指向中位数或者第一个中位数如果数据个数是偶数个。

set,multiset,map的迭代器在插入操作后仍然有效。

代码

class MedianFinder {
    multiset<int> data;
    multiset<int>::iterator mid;
public:
    MedianFinder() {}
    void addNum(int num) {
        const size_t n = data.size();
        data.insert(num);
        if (n == 0) {
            mid = data.begin();
        } else {
            if (num == *mid)
                mid = n & 1 ? mid : next(mid);
            else if (num < *mid)
                mid = n & 1 ? prev(mid) : mid;
            else
                mid = n & 1 ? mid : next(mid);
        }
    }

    double findMedian() { return data.size() & 1 ? *mid : ((*mid) + (double)(*next(mid))) * 0.5; }
};