前言
有时我们需要往 vector 容器中插入 “键值对(pair<int, int>)” 数据,同时又需要按第二个或者第一个进行排序。如上的问题可以借助 STL 的 sort 完成,我们只需要自己写好比较函数。程序可去我的网站GitHub下载。
程序
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> 的第二个值的大小从大往小排序的。