算法複習1 堆

insert() 函數有以下三種用法: 
1、在指定位置loc前插入值爲val的元素,返回指向這個元素的迭代器
2、在指定位置loc前插入num個值爲val的元素 

3、在指定位置loc前插入區間[start, end)的所有元素 

1、題目描述
如何得到一個數據流中的中位數?如果從數據流中讀出奇數個數值,那麼中位數就是所有數值排序之後位於中間的數值。如果從數據流中讀出偶數個數值,那麼中位數就是所有數值排序之後中間兩個數的平均值。

2、代碼
2.1 原理:用兩個堆實現,左邊是做大堆,右邊是最小堆,當數據總數是偶數的時候,中位數是兩個堆首元素和的一半;當數據是奇數的時候,可以約定取最大堆或者是最小堆的首元素。 
2.2 思路:中位數是一串數據中大小排在正中間的數,那麼排到中位數左邊的數都比中位數小,右邊的比中位數大,因此,只需要實時將當前數據分爲兩堆,左邊一堆的最大值比右邊最小堆的最小值要小,中位數就有左邊堆的最大和右邊堆的最小決定。這樣,求中位數只需要實時記錄左右兩堆數據的最大和最小值即可,那麼用最大最小堆即可實現。
--------------------- 
作者:菜鳥程序猿0 
來源:CSDN 
原文:https://blog.csdn.net/u012273127/article/details/79629738 
版權聲明:本文爲博主原創文章,轉載請附上博文鏈接!

class Solution {
public:
    /**
     * @param nums: A list of integers
     * @return: the median of numbers
     */
    priority_queue<int,vector<int>,greater<int>> q; //小頂堆
    priority_queue<int> p;//大頂堆
    vector<int>res;
    vector<int> medianII(vector<int> &nums) {
        // write your code here
        for(int i = 0 ; i < nums.size() ; i++)
        {
            
            
            if(q.size()==0)
            {
                q.push(nums[i]);
                res.push_back(q.top());
            }
             else
            {
                if(p.size() <= q.size())
                {
                    if( nums[i] < q.top())
                    {
                        p.push(nums[i]);
                    }
                    else
                    {
                        p.push(q.top());
                        q.pop();
                        q.push(nums[i]);
                    }
                }
                else
                {
                    if( nums[i] < q.top())
                    {
                        p.push(nums[i]);
                        q.push( p.top());
                        p.pop();
                    }
                    else
                    {
                        q.push(nums[i]);
                    }
                }
                
                res.push_back(p.top());    
                    
            }
            
            
        }
        return res;
    }
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章