6

这是我第一次使用优先队列。我正在尝试为学校实现 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;
   }
};

但是它说这个操作符函数的参数太多了。一段时间以来，我一直在试图弄清楚如何用我的节点管理最小堆优先级队列并一直卡住。有任何想法吗？

4

2 回答 2

7

如果我正确理解您的问题，我相信您真正想要的是制作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);
}
于 2013-03-26T20:13:34.950 回答
2

你需要你的比较函子来实现bool operator()(....)，而不是bool operator<(....)：

struct node_comparison 
{
   bool operator()( const Node* a, const Node* b ) const 
   {
    return a->totalWeight < b->totalWeight;
   }
};
于 2013-03-26T20:14:21.510 回答