c – STL优先级队列和带指针的重载

这是我第一次使用优先级队列.我正在尝试为学校实现Dijkstra的算法,我想我需要一个最小堆才能做到这一点.现在我的节点是指针,我想比较它们的重量,但我认为我不能超载>和<用指针？有没有办法可以做到这一点？代码到目前为止：

priority_queue<Node*,vector<Node*>,node_comparison> minHeap;

然后我有一个结构来比较节点的权重

struct node_comparison 
{
   bool operator<( const Node* a,const Node* b ) const 
   {
    return a->totalWeight < b->totalWeight;
   }
};

但是它说这个运算符功能的参数太多了.我一直试图找出如何用我的节点管理一个最小堆优先级队列一段时间,然后继续卡住.有任何想法吗？

解决方法

如果我正确理解你的问题,我相信你真正想要的是使node_comparison成为一个函子(更具体地说,是一个二元谓词)：

struct node_comparison 
{
    bool operator () ( const Node* a,const Node* b ) const 
    {
        return a->totalWeight < b->totalWeight;
    }
};

仿函数是一个类,其对象提供了调用操作符(operator())的重载,因此可以使用与调用函数相同的语法来调用：

Node* p1 = ...;
Node* p2 = ...;
node_comparison comp;
bool res = comp(p1,p2) // <== Invokes your overload of operator ()

在内部,std :: priority_queue将像我在上面的代码片段中那样或多或少地实例化你的谓词,并以这种方式调用它来执行其元素之间的比较.

仿函数优于常规函数的优点是它们可以保存状态信息(目前您可能不需要这些信息,但这通常是合乎需要的)：

#include <cmath>

struct my_comparator
{
    my_comparator(int x) : _x(x) { }

    bool operator () (int n,int m) const
    {
        return abs(n - _x) < abs(m - _x);
    }

    int _x;
};

例如,上述谓词基于它们与构造时提供的另一个整数的距离来比较整数.这是如何使用它：

#include <queue>
#include <iostream>

void foo(int pivot)
{
    my_comparator mc(pivot);
    std::priority_queue<int,std::deque<int>,my_comparator> pq(mc);

    pq.push(9);
    pq.push(2);
    pq.push(17);

    while (!pq.empty())
    {
        std::cout << pq.top();
        pq.pop();
    }
}

int main()
{
    foo(7);

    std::cout << std::endl;

    foo(10);
}

c – STL优先级队列和带指针的重载

解决方法

猜你在找的C&C++相关文章