前言

        有时我们需要往 vector 容器中插入 “键值对(pair<int, int>)” 数据,同时又需要按第二个或者第一个进行排序。如上的问题可以借助 STL 的 sort 完成。

程序

1. 向算法传递函数

static bool cmp(const pair<int, int>& a, const pair<int, int>& b)
{
    // 以pair对的第2个数的大小从大往小排序 
    return a.second > b.second;
}

2. 借助 lambda 表达式

sort(scores.begin(),scores.end(), 
[](const pair<int, int>& a, const pair<int, int>& b){ return a.second > b.second; }); 
// lambda表达式

3. 完整程序

#include <iostream>
#include <vector>
#include <algorithm>

using  namespace std;

static bool cmp(const pair<int, int>& a, const pair<int, int>& b)
{
    // 以pair对的第2个数的大小从大往小排序 
    return a.second > b.second;
}

int main()
{
    vector<pair<int, int> > scores;
    scores.push_back(make_pair(1, 3));
    scores.push_back(make_pair(2, 1));
    scores.push_back(make_pair(3, 2));
    cout<<"before sort : "<<endl;
    for(int i = 0; i < scores.size(); i++)
    {
        cout<<scores[i].first<<" "<<scores[i].second<<endl;
    }
    //sort(scores.begin(), scores.end(), cmp); // 向算法传递函数 
    sort(scores.begin(),scores.end(), [](const pair<int, int>& a, const pair<int, int>& b){ return a.second > b.second; }); // lambda表达式 
    cout<<"after sort : "<<endl; 
    for(int i = 0; i < scores.size(); i++)
    {
        cout<<scores[i].first<<" "<<scores[i].second<<endl;
    }
    return 0;
}

结果

        下图结果是按 pair<int, int> 的第二个值的大小从大往小降序排序的。

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐