我从 std::priority_queue 派生了一些专门的方法。当我添加一个元素并且队列已满时,其中一种方法是某种固定队列,最小的元素将从队列中删除。
template<typename T,
typename Sequence = std::vector<T>,
typename Compare = std::less<typename Sequence::value_type> >
class fixed_priority_queue : public std::priority_queue<T, Sequence, Compare> {
friend class BCQueue_; // to access maxSize_
public:
fixed_priority_queue(unsigned int maxSize)
: maxSize_(maxSize) {}
void insertWithOverflow(const T& x) {
if (this->size() == maxSize_) {
auto beg = this->c.begin();
auto end = this->c.end();
auto min = std::min_element(beg, end);
if(x > *min) {
*min = x;
std::make_heap(beg, end);
}
}
else {
this->push(x);
}
}
// ...
private:
fixed_priority_queue() {}
const unsigned int maxSize_;
};
这是我使用的比较器:
class ScoreLessThan : public std::binary_function<
std::shared_ptr<CCandidate>, std::shared_ptr<CCandidate>, bool> {
public:
bool operator()(
const std::shared_ptr<CCandidate>& a, const std::shared_ptr<CCandidate>& b) const {
return a->score > b->score;
}
};
我包装了我派生的 fixed_priority_queue 类以保持功能有点分离,所以最后我有了这个:
class BCQueue_ : public fixed_priority_queue<
std::shared_ptr<CCandidate>, std::vector<std::shared_ptr<CCandidate> >, ScoreLessThan> {
public:
BCQueue_(size_t maxSize)
: fixed_priority_queue(maxSize) {}
bool willInsert(float score) {
return size() < maxSize_ || top()->score < score;
}
};
我可以这样使用:
BCQueue_ queue(30);
这CCandidate
只是一些数据的持有者。一个属性是score
我在上面的比较器中使用的字段。
当我将上面的类与 CCandidate 一起用作原始指针时,所有编译都有问题并且工作正常,现在我想用std::shared_ptr
(就像我上面所做的那样)替换原始指针,我得到一个编译错误:
... 199:5: error: no match for ‘operator>’ in ‘x >min.__gnu_cxx::__normal_iterator::operator* [with _Iterator = std::shared_ptr*, _Container = std::vector >, __gnu_cxx::__normal_iterator::reference = std::shared_ptr&]()’
...
... :199:5: note: candidates are:
... :199:5: note: operator>(int, int)
... :199:5: note: no known conversion for argument 2 from ‘std::shared_ptr’ to ‘int’
也许这是一个简单的问题。我不确定我是否正确定义了比较器,或者我是否需要更改比较insertWithOverflow()
x > *min
,实际上我不知道我应该在那里更改什么。
我应该提一下,我在 stackoverflow 上找到了 `insertWithOverflow' 的实现,它正好符合我的需要。见这里:如何使 STL 的 priority_queue 固定大小
就像说的那样,使用原始指针,这一切都没有问题。有人可以帮我解决这个问题。提前致谢!