Мне нужно вычислить медиану. Мне сказали, что лучший способ сделать это в этом конкретном приложении — с приоритетной очередью. Я понятия не имею, как действовать дальше. Я был бы очень признателен за любую помощь.
Вычислить медиану, используя приоритетные очереди
Ответы (1)
Это должно сделать это. Поддерживайте две очереди с приоритетом чисел больше и меньше медианного значения. Сместите значения между двумя очередями так, чтобы они оставались сбалансированными или близкими к сбалансированным, и определите медиану на основе верхних значений приоритетных очередей.
#include <queue>
#include <vector>
#include <functional>
#include <limits>
#include <iostream>
template<class T>
class RunningMedian {
private:
//Values greater than the median sorted so that smallest is on top
std::priority_queue<T, std::vector<T>, std::greater<T> > upper;
//Values smaller than the median sorted so that greatest is on top
std::priority_queue<T, std::vector<T>, std::less<T> > lower;
public:
RunningMedian(){
//Seed the queues
upper.push(std::numeric_limits<T>::max());
lower.push (std::numeric_limits<T>::min());
}
void push(T val){
//Add number to the property queue
//If number is greater than the least upper number, it is an upper number
if(val>=upper.top())
upper.push(val);
else //Otherwise it is a lower number
lower.push(val);
//Rebalance
//If the queues sizes differ by 2, they need to be rebalanced so that they
//differ by 1.
if(upper.size()-lower.size()==2){ //Upper queue is larger
lower.push(upper.top()); //Move value from upper to lower
upper.pop(); //Remove same value from upper
} else if(lower.size()-upper.size()==2){ //Lower queue is larger
upper.push(lower.top()); //Move value from lower to upper
lower.pop(); //Remove same value
}
}
T getMedian() const {
if(upper.size()==lower.size()) //Upper and lower are same size
return(upper.top()+lower.top())/((T)2.0); //so median is average of least upper and greatest lower
else if(upper.size()>lower.size()) //Upper size greater
return upper.top();
else //Lower size greater
return lower.top();
}
};
int main(){
RunningMedian<int> rm;
rm.push(10);
rm.push(3);
rm.push(1);
rm.push(20);
rm.push(5);
rm.push(8);
rm.push(-1);
std::cout<<rm.getMedian()<<std::endl; //Gives 5
}
person
Richard
schedule
04.03.2015
Имеет смысл. Огромное спасибо!
- person Zechariah Schneider; 05.03.2015
Не за что. Если вы чувствуете, что это ответило на ваш вопрос, не стесняйтесь принять ответ. В противном случае дайте мне знать, если какая-либо часть вашего вопроса останется без ответа.
- person Richard; 05.03.2015
std::priority_queue<>
- person WhozCraig   schedule 04.03.2015