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

时间:2015-03-09 10:42:12

标签: c++ graph stl priority-queue

我正在尝试对图形边缘进行排序,但我无法这样做。 下面给出了我实现优先级队列的方法。

  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();
}

2 个答案:

答案 0 :(得分: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; 
              });

答案 1 :(得分:0)

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

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

应该是

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

注意左边第二个索引的名称。 的 Live On Coliru