Leetcode刷题 - 在数据流找中位数（Find Median from Data Stream）

方法一：Insertion Sort

 1 class MedianFinder {
 2     vector<int> store; // resize-able 
 3 public:
 4     /** initialize your data structure here. */
 5     MedianFinder() {
 6         
 7     }
 8     
 9     void addNum(int num) {
10         if (store.empty())
11             store.push_back(num);
12         else
13             // insert(position, value)
14             // lower_bound(interator first, interator last, num) -> return 
15             // return 在range里第一个小于这个num的位置
16             // binary search combined with insertion
17             store.insert(lower_bound(store.begin(), store.end(), num), num);
18     }
19     
20     double findMedian() {
21         int n = store.size();
22         //利用二进制快速判断奇偶数
23         return n & 1 ? store[n/2] : ((double) store[n/2 - 1] + store[n/2])*0.5;
24     }
25 };

方法二：Two Heap

 1 class MedianFinder{
 2     //优先队列建立最大堆
 3     priority_queue<int> lo; // max heap
 4     // 优先队列建立最小堆
 5     priority_queue<int, vector<int>, greater<int>> hi; // min heap
 6 public:
 7     void addNum(int num){
 8         // Add to max heap
 9         lo.push(num);
10         // balance step
11         hi.push(lo.top());
12         
13         lo.pop();
14         
15         // maintain size property
16         if(lo.size() < hi.size()){
17             lo.push(hi.top());
18             hi.pop();
19         }
20     }
21     
22     double findMedian(){
23         return lo.size() > hi.size() ? lo.top() : ((double) lo.top() + hi.top())*0.5;
24     }
25 };

 

posted @ 2020-09-11 17:26  cancantrbl  阅读(231)  评论(0)    收藏  举报