《剑指offer》-数据流中的中位数

简介: 如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。最开始的思路就是用map或者set存储。

如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。

最开始的思路就是用map或者set存储。习惯写python就想直接用median的key去访问median,但是C++ STL的map或者set没有key这个东西,如果用迭代器那么访问元素复杂度是O(n)

看到很多解法是用两个堆来做,一个最大堆,一个最小堆,一开始不理解。后来发现这样的好处是把数据总体切分为两部分,一部分(最大堆)所有元素都比另一部分(最小堆)小。然后当有新元素需要insert的时候,根据现有元素总数奇偶,决定先压入哪个堆,然后弹出一个元素,弹出元素放入另一个堆。

最后的答案处理,根据元素总数奇偶,决定从两个堆分别取还是从特定的那个取。

class Solution{
public:
    void Insert(int num){
        if (maxS.size() == minS.size()){
            maxS.insert(num);
            minS.insert(*maxS.begin());
            maxS.erase(maxS.begin());
        }
        else{
            minS.insert(num);
            maxS.insert(*minS.begin());
            minS.erase(minS.begin());
        }
    }

    double GetMedian(){
        int num = maxS.size() + minS.size();
        
        double median;
        if ((num&1)==1){
            median = *minS.begin();
        }
        else{
            median = (*maxS.begin() + *minS.begin()) / 2.0;
        }
        return median;
    }
private:
    multiset<int, greater<int> > maxS;
    multiset<int, less<int> > minS;
};
目录
相关文章
|
6月前
|
算法 测试技术 C++
【动态规划】【前缀和】【数学】2338. 统计理想数组的数目
【动态规划】【前缀和】【数学】2338. 统计理想数组的数目
|
算法 C++
剑指offer(C++)-JZ41:数据流中的中位数(算法-排序)
剑指offer(C++)-JZ41:数据流中的中位数(算法-排序)
剑指offer(C++)-JZ41:数据流中的中位数(算法-排序)
|
6月前
|
C++
剑指 Offer 41:数据流中的中位数
剑指 Offer 41:数据流中的中位数
28 0
|
算法 测试技术 C#
C++算法:数据流的中位数
C++算法:数据流的中位数
|
存储
剑指offer 42. 数据流中的中位数
剑指offer 42. 数据流中的中位数
41 0
|
存储 算法 Java
数据流中的中位数,我轻敌了
最近面试时候遇到一个非常有意思的hard题,面试官没让写代码让说思路,但放在正常应届生招聘那可能就要手撕了,在剑指offer的第41题和力扣【数据流中的中位数】。
145 0
数据流中的中位数,我轻敌了
【算法题解】拓扑序计数+树形DP
【算法题解】拓扑序计数+树形DP
【算法题解】拓扑序计数+树形DP
|
机器学习/深度学习 存储 人工智能
LOJ6285.数列分块入门 9(分块在线求区间众数)
LOJ6285.数列分块入门 9(分块在线求区间众数)
134 0
再学一道算法题: 两个有序序列的中位数
再学一道算法题: 两个有序序列的中位数
再学一道算法题: 两个有序序列的中位数