本题源自leetcode
---------------------------------------------------
思路 用俩个优先队列存储。 优先队列 是按降序排列的
1 当 加入一个数字时。先把数字加入小的队列中。然后弹出小队列的队头,并把其相反数压入大的队列。如果小队列的个数小于大队列。就把大的队列头弹栈,取反后压入小的。
2 这样就保证了,如果小的个数大于大的个数,则为奇数,切中位数在小队列的队头。否则就是偶数。中位数是小堆头加上取反的大队头求平均数
代码:
class MedianFinder {
priority_queue<int> small,large;
public:
/** initialize your data structure here. */
MedianFinder() {
}
void addNum(int num) {
small.push(num);
large.push(-small.top());
small.pop();
if (small.size() < large.size()) {
small.push(-large.top());
large.pop();
}
}
double findMedian() {
return small.size() > large.size()
? small.top()
: (small.top() - large.top()) / 2.0;
}
};