原题:
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.
For example:
addNum(1) addNum(2) findMedian() -> 1.5 addNum(3) findMedian() -> 2
解决方法:
用set保存加入的值,low,high分别指向中间两个数的位置。
- 如果原来set是空,low,high都指向起始点。
- 如果原来的个数是奇数,加入新的数后个数变成偶数,low和high指向不同的数。如果加入的数比原来最小的还小,表明是加入到前面,只需要改变low的位置,否则改变high的位置。
- 如果原来的个数是偶数,加入新的数后个数变成奇数,low和high应该指向同一个数。如果加入的数在low和high之间,则low和high都需要被改变指向这个数;如果加入的是一个小数,则减少high使之等于low;如果加入的是一个大一点的数,则增加low使之等于high。
代码:
class MedianFinder {
multiset<int> s;
multiset<int>::iterator low, high;
public:
/** initialize your data structure here. */
MedianFinder() {
}
void addNum(int num) {
int size = s.size();
s.insert(num);
if (size == 0){
low = high =s.begin();
}else if (size % 2){
if (num < *low)
--low;
else
++high;
}else{
if (num > *low && num < *high){
++low;
--high;
}else if ( num <= *low){
low = --high;
}else{
++low;
}
}
}
double findMedian() {
return (*low + *high) * 0.5;
}
};