实时中位数

题目描述
现有一些随机生成的数字要将其依次传入,请设计一个高效算法,对于每次传入一个数字后,算出当前所有传入数字的中位数。(若传入了偶数个数字则令中位数为第n/2小的数字,n为已传入数字个数)。

使用优先队列创建一个最大堆和一个最小堆,最大堆存放较小的元素,最小堆存放较大的元素,且始终保持最大堆的元素数目等于最小堆的元素数目或者比最小堆的元素数目多1,则最大堆的堆顶即为中位数。

 1 class Middle {
 2 public:
 3     vector<int> getMiddle(vector<int> A, int n) {
 4         priority_queue<int, vector<int>, greater<int>> minh;
 5         priority_queue<int, vector<int>, less<int>> maxh;
 6         vector<int> res;
 7         for (int i = 0; i < n; ++i) {
 8             if (maxh.empty()) maxh.push(A[i]);
 9             else if (A[i] < maxh.top()) {
10                 maxh.push(A[i]);
11                 if (maxh.size() - minh.size() > 1) {
12                     minh.push(maxh.top());
13                     maxh.pop();
14                 }
15             } else {
16                 minh.push(A[i]);
17                 if (minh.size() > maxh.size()) {
18                     maxh.push(minh.top());
19                     minh.pop();
20                 }
21             }
22             res.push_back(maxh.top());
23         }
24         return res;
25     }
26 };

 

posted @ 2019-07-25 09:55  ppwq  阅读(222)  评论(0编辑  收藏  举报