Leetcode 295. 数据流的中位数 —— 终于刷完LC Top 100

题目

Leetcode 295. 数据流的中位数

代码（首刷看解析）

class MedianFinder {
public:
    priority_queue<int> q_max;
    priority_queue<int, vector<int>, greater<int>> q_min;   // 小根堆
    MedianFinder() {

    }
    
    void addNum(int num) {
        if(q_min.size() >= q_max.size()) {  // 如果小根堆更多，则将小根堆顶给大根堆 保持两个队列“高度平衡”
            q_max.push(num);
            q_max.push(q_min.top());    // 堆顶交换是为了保证两小根堆的始终比大根堆的小
            q_min.pop();
        } else {
            q_min.push(num);
            q_min.push(q_max.top());
            q_max.pop();
        }
    }
    
    double findMedian() {
        if(q_min.size() < q_max.size()) {   // max的始终会比min的多一个
            return q_max.top();
        } else if(q_min.size() > q_max.size()) {
            return q_min.top();
        } else {
            return (q_min.top() + q_max.top()) / 2.0;
        }
    }
};

代码（8.22 二刷看解析）

BM48 数据流中的中位数

1. 首先需要保证两个堆数量高度平衡
2. 保证大顶堆的都比小顶堆的小，所以需要进行堆顶交换push.

#include <functional>
#include <queue>
class Solution {
public:
    void Insert(int num) {
        if(minQ.size() == maxQ.size()) {
            minQ.push(num); // 首先要保证两个堆元素数量高度平衡
            maxQ.push(minQ.top());  // 小根堆的最小元素给大根堆，保证大根堆都比小根堆小
            minQ.pop();
        } else {
            maxQ.push(num);
            minQ.push(maxQ.top());  // 大根堆的最大元素给小根堆，保证小根堆都比大根堆大
            maxQ.pop();
        }
    }

    double GetMedian() { 
        if((minQ.size()+maxQ.size())%2==1)
            return maxQ.top();
        return (maxQ.top()+minQ.top())/2.0;
    }
    priority_queue<int> maxQ;
    priority_queue<int, vector<int>, greater<int>> minQ;
};

代码（9.13 三刷自解）

class MedianFinder {
public:
    priority_queue<int> maxQ;
    priority_queue<int,vector<int>, greater<int>> minQ;
    MedianFinder() {}
    
    void addNum(int num) {
        if(maxQ.size() <= minQ.size()) {
            minQ.push(num);
            maxQ.push(minQ.top());
            minQ.pop();
        } else {
            maxQ.push(num);
            minQ.push(maxQ.top());
            maxQ.pop();
        }
    }
    
    double findMedian() {
        if((maxQ.size()+minQ.size())%2==1)
            return maxQ.top();
        return (maxQ.top()+minQ.top())/2.0;
    }
};

代码（9.20 四刷自解）

class MedianFinder {
public:
    priority_queue<int> maxQ;
    priority_queue<int, vector<int>, greater<int>> minQ;
    MedianFinder() {}
    
    void addNum(int num) {
        if(maxQ.size() >= minQ.size()) {
            maxQ.push(num);
            minQ.push(maxQ.top());
            maxQ.pop();
        } else {
            minQ.push(num);
            maxQ.push(minQ.top());
            minQ.pop();
        }
    }
    
    double findMedian() {
        if((maxQ.size()+minQ.size())%2) {
            return minQ.top();
        } else {
            return (maxQ.top()+minQ.top())/2.0;
        }
    }
};