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表达式获取外部变量的方式如下: