在c++中,由于sort()
函数默认提供的是由小到大的排序方式,因此有时候我们需要自定义排序规则来实现由大到小的排序。
一维vector<>排序
#include <bits/stdc++.h>
using namespace std;
void printvec(vector<int>& vec){
for(int i = 0; i < vec.size(); i++){
cout << vec[i] << " ";
}
}
int main(){
int num[] = {1, 9, 3, 8, 5, 7, 12};
vector<int> vec(num, num+(sizeof(num)/sizeof(num[0])));
//使用sort函数进行排序,默认为升序
sort(vec.begin(), vec.end(), [](int a, int b){return a > b;});
printvec(vec);
return 0;
}
输出的结果就是:12 9 8 7 5 3 1
二维vector<>排序 (以某一位的大小进行排序)
#include <bits/stdc++.h>
using namespace std;
void printvec(vector<vector<int>>& vec){
for(int i = 0; i < vec.size(); i++){
for(int j = 0; j < vec[0].size(); j++){
cout << vec[i][j] << " ";
}
}
}
int main(){
vector<vector<int>> vec{{1, 9, 3},{8, 5, 7}, {12, 13, 14}};
sort(vec.begin(),vec.end(),[](vector<int> a, vector<int> b){return a[1] < b[1];});
cout << endl;
printvec(vec);
return 0;
}
// 优先按照第1位排,若值相等,则按照下一位排
int main(){
vector<vector<int>> vec{{1, 9, 3},{8, 5, 7}, {12, 13, 14}};
sort(vec.begin(),vec.end(),[](vector<int> a, vector<int> b){
for (int col = 0; col < 3; ++col) {
if (a[col] != b[col]) return a[col] < b[col];
}
return false;
});
cout << endl;
printvec(vec);
return 0;
}
输出结果为:8 5 7 1 9 3 12 13 14
注意!sort()中的lambda函数中的参数请使用&引用,可以节省时间。
在做lc 435. 无重叠区间 题目时,若不用&会导致超时。
附上可以通过的代码:
class Solution {
public:
int eraseOverlapIntervals(vector<vector<int>>& intervals) {
sort(intervals.begin(), intervals.end(), [](vector<int> &a, vector<int> &b)
{
return a[1] < b[1];
});
int numbRemove = 0;
int prev = intervals[0][1];
for (int i = 1; i < intervals.size(); ++i) {
if (intervals[i][0] < prev) {
++numbRemove;
}
else {
prev = intervals[i][1];
}
}
return numbRemove;
}
};