lambda自定义比较规则-sort函数或优先队列

发布时间:2023年12月18日

Lambda表达式的一般形式为

[captures](params){body}

对于优先队列的自定义排序规则,常见方法是写成结构体形式

struct cmp{
	bool operator()(pair<int,int> map1,pair<int,int> map2){
		return map1.second>map2.second;
	}
};
priority_queue<pair<int,int>,vector<pair<int,int>>,cmp> que;

写成lambda表达式形式有两个注意点,一个是要用decltype声明,并且实例化时也要注意

auto cmp=[](pair<int,int> map1,pair<int,int> map2){return map1.second>map2.second;};
priority_queue<pair<int,int>,vector<pair<int,int>>, decltype(cmp)> que(cmp);

整体测试代码如下:

#include<bits\stdc++.h>
//#include<unordered_map>
//#include<priority_queue>
using namespace std;
struct cmp{
	bool operator()(pair<int,int> map1,pair<int,int> map2){
		return map1.second>map2.second;
	}
};
int main(){
	unordered_map<int,int> map;
	map[0]=3;
	map[1]=2;
	map[2]=6;
	map[3]=7;
	map[4]=8;
	//priority_queue<pair<int,int>,vector<pair<int,int>>,cmp> que;
	auto cmp=[](pair<int,int> map1,pair<int,int> map2){return map1.second>map2.second;};
	priority_queue<pair<int,int>,vector<pair<int,int>>, decltype(cmp)> que(cmp);
	for(auto it:map){
		if(que.size()>3){
			que.pop();
		}
		que.push(it);
	}
	while(!que.empty()){
		cout<<que.top().second<<" ";
		que.pop();
	}
	system("pause");
}

对于sort函数而言,常见做法

bool cmp(int a,int b){
	return a>b;
}
sort(nums.begin(),nums.end(),cmp);

写成lambda表达式:

sort(nums.begin(),nums.end(),[](int a,int b){return a>b;});

如果在自定义函数内要用到外部定义的变量,那么lambda表达式更加合适
比如我们想根据对某个数组进行排序,并且想知道排序完后每个位置的数值对应原来的下标。就需要在自定义比较函数内部获取外部变量

#include<bits\stdc++.h>
using namespace std;
int main(){
	vector<int> nums{1,3,4,6,2,9};
	vector<int> index(nums.size());
	for(int i=0;i<nums.size();i++){
		index[i]=i;
	}
	//对索引数组进行排序,排序的依据是索引对应nums数组中的值
	sort(index.begin(),index.end(),[&](int i, int j){return nums[i]<nums[j];});
	//其中&代表以引用的方式获取外部变量 
	for(int i:index){
		cout<<i<<" ";
	}
	system("pause");
}

Lambda表达式获取外部变量的方式如下:
在这里插入图片描述

文章来源:https://blog.csdn.net/weixin_45850570/article/details/135073059
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。