用于对图形边缘进行排序的C ++ Priority Queue实现

Question

我试图排序图形边缘但我无法这样做。 下面给出的是我实现的优先级队列实现相同。

  class CompareDistance
    {
        public:    
            bool operator()(pair<int,int> n1,pair<int,int> n2)
            {
                if(g[n1.first][n2.second] < g[n2.first][n2.second])
                return true;
                else
                return false;
           }
    };
    int g[3][3] = {{4,1,3},{3,3,3},{3,3,3}};//graph edges


int main()
{

    priority_queue<pair<int,int>,vector<pair<int,int> >,CompareDistance> pq;


    for(int i = 0 ; i < 3 ; i++)
        for(int j = 0 ; j < 3 ; j++)
            pq.push(pair<int,int>(i,j));

    cout<<"\t"<<g[pq.top().first][pq.top().second];//does not give the correct result
    pq.pop();




    getch();
}

Answer 1

好的，如果我理解正确，你就会有一个有向图（就像三个节点的示例完整图一样）。 每个弧都有相关的值 - 权重，你需要的是按这个值对弧的集合进行排序。 首先，您应该创建弧的集合。 我不认为priority_queue是你最好的选择。 我会在这里使用向量因为简单：

std::vector<std::pair<int, std::pair<int, int>>> arcs;

其中第一对将包含弧重和弧之间存在的有向节点对。 在添加所有节点之后，您只需对指定要比较的自定义函数的集合进行排序。 如果你使用c ++ 11，它可能如下所示：

std::sort(arcs.begin(), arcs.end(), 
          [] (const std::pair<int, std::pair<int, int>>& a,
              const std::pair<int, std::pair<int, int>>& b) { 
                  return a.first < b.first; 
              });

Answer 2

您的代码无法编译，但这是一个简单的修复。 我假设g包含边权重。 这里有一个错误：

if(g[n1.first][n2.second] < g[n2.first][n2.second])

它应该是

if(g[n1.first][n1.second] < g[n2.first][n2.second])

记住左边第二个索引的名称。 住在Coliru