我正在使用随机访问实现改进的priority_queue。
template <class T, class Container = std::vector<T> >
class Heap {
public:
Heap() {}
Heap(const Container& container) {
container_ = container;
std::make_heap(container_.begin(), container_.end());
}
Heap<T, Container>& operator=(const Heap<T, Container>& heap) {
if (this != &heap)
container_ = heap.container_;
return *this;
}
void push(const T& x) {
container_.push_back(x);
std::push_heap(container_.begin(), container_.end());
}
void pop() {
std::pop_heap(container_.begin(), container_.end());
container_.pop_back();
}
const T& top() {
return container_.front();
}
const Container& getContainer() const {
return container_;
}
T& operator[](size_t n) {
return container_[n];
}
typename Container::const_iterator begin() const {
return container_.begin();
}
typename Container::const_iterator end() const {
return container_.end();
}
size_t size() const {
return container_.size();
}
T& base() {
return container_.back();
}
Container::iterator erase(Container::iterator position) {
return container_.erase(position);
}
private:
Container container_;
};
我采取了正确的方式吗?
答案 0 :(得分:1)
对我来说看起来不那么好:
getContainer()
方法显示界面不清晰 - 为什么你只是简单地公开实现细节?答案 1 :(得分:0)
你的pop()方法可能违反堆排序。使用pop_heap()而不是pop_back()。我现在似乎无法找到任何其他错误。
您可以通过逐个推送随机整数和pop()来轻松测试这样的实现。你应该按排序顺序取回它们。这称为堆排序。
建议:
您可以为此类实现一个const迭代器,而不是将ref返回容器。
请注意,您不应更改随机访问的元素的键,因为它可能会破坏堆结构。如果您需要这样的功能,您应该实现一个change_key函数,它可以安全地更改密钥并保持堆排序。