c++ stl::priority queue 使用 2 个键排序不正确答案

【问题标题】：cpp stl::priority queue using 2 keys not sorting correctlyc++ stl::priority queue 使用 2 个键排序不正确
【发布时间】：2018-11-01 03:05:53
【问题描述】：

我已经定义了一个结构如下：

using namespace std; 
struct Person {
    int id; 
    int age; 
    string country_of_origin; 

};

我正在尝试使用 STL 优先级队列类根据 id 按升序对人员进行排序，并根据原籍国按字典顺序对匹配 id 的值进行排序。以下是我的代码：

#include <queue> 
#include <vector> 
#include <iostream>
#include <queue>
#include <string> 

using namespace std; 
struct Person {
    int id; 
    int age; 
    string country_of_origin; 

};

struct PersonCompare
{
    bool operator()(const Person &p1, const Person &p2) const
    {
        if (p1.id>p2.id) {
            return true; 
        } else if (p1.country_of_origin>p2.country_of_origin){
            return true; 
        }  else {
            return false; 
        }
    }
};




int main(int argc, char ** argv)
{
    Person p={1,2,"Zimbabwe"}; 
    Person q={3,4,"New Zealand"};
    Person r={5,7,"Wales"}; 
    Person s={2,3,"India"};
    Person t={1,4,"Sri Lanka"}; 
    Person u={1,4,"Benin"}; 
    priority_queue<Person,vector<Person>,PersonCompare> queue;
    queue.push(p); 
    queue.push(q); 
    queue.push(r); 
    queue.push(s); 
    queue.push(t); 
    queue.push(u);
    while (!queue.empty())
    {
        Person p = queue.top();
        cout << p.id << "\t"<<p.age<<"\t"<<p.country_of_origin<<endl;  
        queue.pop();
    }

}

我得到的输出是：

1       4       Benin
2       3       India
3       4       New Zealand
1       4       Sri Lanka
1       2       Zimbabwe
5       7       Wales

代替：

1       4       Benin
1       4       Sri Lanka
1       2       Zimbabwe
2       3       India
3       4       New Zealand
5       7       Wales

为什么会这样？

【问题讨论】：

else if (p1.country_of_origin>p2.country_of_origin) 应该是else if (p1.id == p2.id && p1.country_of_origin > p2.country_of_origin)

标签： c++ stl priority-queue

【解决方案1】：

您可以通过对第一个字段进行排序来进行字典比较，只有当它们相等时，然后测试第二个字段以打破平局。如果按如下方式重写比较器，它应该可以工作：

struct PersonCompare
{
    bool operator()(const Person &p1, const Person &p2) const
    {
        if (p1.id > p2.id) {
            return true; 
        } else if (p1.id < p2.id) {
            return false;
        }
        return p1.country_of_origin > p2.country_of_origin;
    }
};

另一种方法是使用std::tie，它更简洁，并且可以正确地进行字典比较：

#include <tuple>

struct PersonCompare
{
    bool operator()(const Person &p1, const Person &p2) const
    {
        return std::tie(p1.id, p1.country_of_origin) > std::tie(p2.id, p2.country_of_origin);
    }
};

【讨论】：