自定义排序
sort函数第三个参数compare,为自定义比较函数指针,原型如下:
bool cmp(const Type &a, const Type &b);
如果是想升序,那么就定义当a<b的时候返回true;
如果是想降序,那么就定义当a>b的时候返回true;
注意compare函数写在类外或者定义为静态函数
std::sort要求函数对象,或是静态/全局函数指针,非静态成员函数指针不能直接传递给std::sort。
示例
bool cmp(const pair<int,int> &a, const pair<int,int> &b)
{
return a.second>b.second;
}
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
vector<int> res;
unordered_map<int, int> countMap;
for (auto item:nums) {
countMap[item]++;
}
vector<pair<int, int>> countVec(countMap.begin(), countMap.end());
// sort(countVec.begin(), countVec.end(), [](const pair<int,int> &a, const pair<int,int> &b){
// return a.second>b.second;
// });
sort(countVec.begin(), countVec.end(), cmp);
for (int i=0;i<k;i++) {
res.push_back(countVec[i].first);
}
return res;
}
};