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优先级队列和指针重载，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/15646451/