题目如下：(https://leetcode.com/problems/find-median-from-data-stream/)
Median is the middle value in an ordered integer list. If the size of the list is even, there is no middle value. So the median is the mean of the two middle value.
Examples:
[2,3,4] , the median is 3
[2,3], the median is (2 + 3) / 2 = 2.5
Design a data structure that supports the following two operations:
- void addNum(int num) - Add a integer number from the data stream to the data structure.
- double findMedian() - Return the median of all elements so far.
题目需要实现两个函数：向一个整数表中添加元素以及找中位数。以下是两个思路：
思路一
如果用vector储存整数，找中位数较容易（O(1)），添加整数可能会较耗时间，于是尝试使用时间复杂度为O(log(n))的二分插入。
class MedianFinder {
public:
bool odd; //长度是否为奇数。奇数为true，偶数为false
int begin, end, mid; //数组头、尾、中间的下标
vector<int> l; //数组
MedianFinder(): odd(false), begin(), end(), mid() {}
//添加整数
void addNum(int num) {
//更新
odd = odd ? false : true;
begin = ;
end = l.size() - ;
//二分查找
while (begin <= end)
{
mid = (begin + end) / ;
if (l[mid] == num)
{
break;
}
else if (l[mid] > num)
{
end = mid - ;
}
else
{
begin = mid + ;
}
}
//插入
if (begin > end)
{
l.insert(l.begin()+begin, num);
}
else
{
l.insert(l.begin()+mid, num);
}
}
//返回中位数
double findMedian() {
begin = ;
end = l.size() - ;
mid = (begin + end) / ;
if (odd)
{
return l[mid];
}
else
{
return ((double)l[mid] + l[mid+]) / ;
}
}
};
思路二
题目的提示说要用堆，于是考虑stl中的优先队列。可以将中位数两侧的数据分别储存至两个优先队列，一个整数大的优先级高（默认）（相当于排好序的大顶堆），另一个整数小的优先级高（相当于排好序的小顶堆）。
class MedianFinder {
public:
bool odd; //长度是否为奇数。奇数为true，偶数为false
priority_queue<int> front; //数组的前半部分，优先队列默认为较大的优先级高
priority_queue<int, vector<int>, greater<int>> back; //数组的后半部分，较小的数优先级高
MedianFinder(): odd(false) {}
//添加整数
void addNum(int num) {
//更新长度的状态
odd = odd ? false : true;
//插入num，并保证front长度不小于back长度
if (odd)
{
if (back.size() && num > back.top())
{
back.push(num);
front.push(back.top());
back.pop();
}
else
{
front.push(num);
}
}
else
{
if (front.size() && num > front.top())
{
back.push(num);
}
else
{
front.push(num);
back.push(front.top());
front.pop();
}
}
}
//返回中位数
double findMedian() {
if (odd)
{
return front.top();
}
else
{
return ((double)front.top() + back.top()) / ;
}
}
};
附：
//先留个坑，考完试再填吧……