前言

有时我们需要往 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> 的第二个值的大小从大往小排序的。

Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐