查找无序数组的中位数

一、思路

中位数无非就是在一组数最中间的位置，或者中间的两个数的平均值，可以想象利用一个大顶堆在左边，小顶堆在右边，则左边的堆顶元素是比较小的数的最大值，而右边大顶堆的堆顶元素是比较大的数的最小值，那么中位数就夹在两个堆顶元素的中间，也就是两个堆顶元素的平均值（数组长度为偶数，两个堆大小相等），或是元素数量比较多的堆的堆顶元素（数组长度为奇数，两个堆大小自然不相等，多出来的那个堆的堆顶元素自然就是中位数）。

具体实现：遍历数组中的每个元素，将逐个元素分别加入大顶堆或小顶堆（若当前元素进入大顶堆排好序后将堆顶元素交给小顶堆，则下一个元素进入小顶堆排好序后将堆顶元素交给大顶堆，如此循环直至遍历整个数组）。利用一个大顶堆和一个小顶堆互相交换堆顶元素，实现每轮大顶堆中的最大值交给右边的小顶堆，或小顶堆的最小值交给左边的大顶堆，从而将中位数夹住。完成查找无序数组的中位数。

二、代码

#include <iostream>
#include <queue>
using namespace std;

int SearchMiddle(int c[], int len){         // 查找中位数函数
    priority_queue<int, vector<int>, less<int> >  maxpq;     // 声明一个最大优先队列
    priority_queue<int, vector<int>, greater<int> > minpq;    // 声明一个最小优先队列

    for(int i = 0; i < len; ++i){
        if(maxpq.size() == minpq.size()){      // 若大顶堆长度等于小顶堆长度，则将小顶堆的堆顶元素给大顶堆
            minpq.push(c[i]);
            maxpq.push(minpq.top());
            minpq.pop();
        }
        else{                     // 若大小顶堆长度不相等，则将大顶堆的堆顶元素给小顶堆
            maxpq.push(c[i]);
            minpq.push(maxpq.top());
            maxpq.pop();
        }
    }

    // 如果数组长度为偶数，返回两个堆顶元素之和的平均值，若为奇数，则返回大顶堆的队头元素（因为最后一个入队的元素会加入大顶堆）
    return len % 2 == 0 ? (maxpq.top() + minpq.top()) / 2.0 : maxpq.top();
}


int main()
{
    int a[] = {5, 6, 1, 1, 3, 4, 7, 8, 3, 9};
    int len = sizeof(a) / sizeof(a[0]);         // 数组长度
    int m = SearchMiddle(a, len);               // 中位数
    cout << m << endl;
    return 0;
}