C++ sort函数中利用lambda进行自定义排序规则

在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;
}

输出结果为：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;
    }
};