C++ vector 自定义排序规则（vector＜vector＜int＞＞、vector＜pair＜int,int＞＞）

vector< int >

cpp 复制代码

vector<int> vec{1,2,3,4};

//默认从小到大排序  1234
sort(vec.begin(),vec.end());
//从大到小排序 4321
sort(vec.begin(),vec.end(),greater<int>());

二维向量vector<vector< int >>

cpp 复制代码

vector<vector<int>> vec{{0,2},{1,5},{1,9},{4,6},{5,9},{8,10}};

//默认优先对第一元素进行从小到大排序，第一元素相同的，按照第二元素从小到大排序
sort(vec.begin(),vec.end()); //[0,2],[1,5],[1,9],[4,6],[5,9],[8,10]

定义cmp函数

1.外部定义

平时的编译器里面自己写的时候 cmp函数前面加不加static都行，但是leetcode刷题时，代码还有实现都是写在类里面的，cmp函数前面一定要加上static修饰。因此建议所有的返回类型都为static bool 类型。
这里不能使用C++11中的auto关键词进行自动类型推导
C++ sort函数自定义cmp函数中参数带&符号

cpp 复制代码

static bool cmp(const vector<int>& v1, const vector<int>& v2){
    //如果第一元素相等，则比较第二元素
    if (v1[0] == v2[0])
        return v1[1] < v2[1];
    return v1[0] > v2[0];
}

vector<vector<int>> vec{{0,2},{1,5},{1,9},{4,6},{5,9},{8,10}};
sort(vec.begin(),vec.end(),cmp);
//[8,10],[5,9],[4,6],[1,5],[1,9],[0,2]

2.匿名函数Lambda

cpp 复制代码

vector<vector<int>> vec{{0,2},{1,5},{1,9},{4,6},{5,9},{8,10}};

sort(vec.begin(),vec.end(),[&](const vector<int> &v1, const vector<int> &v2){
    //如果第一元素相等，则比较第二元素
    if (v1[0] == v2[0])
        return v1[1] < v2[1];
    return v1[0] > v2[0];
});
//[8,10],[5,9],[4,6],[1,5],[1,9],[0,2]

可以使用auto

cpp 复制代码

sort(vec.begin(),vec.end(),[&](const auto &v1, const auto &v2){
    //如果第一元素相等，则比较第二元素
    if (v1[0] == v2[0])
        return v1[1] < v2[1];
    return v1[0] > v2[0];
});

vector＜pair＜int, int＞＞

初始化

cpp 复制代码

vector<pair<int,int>> vec;

1. vec.emplace_back(1,2);
2. 能够直接用map转化
//unordered_map<int,int> map;
vector<pair<int,int>> vec(map.begin(),map.end());

cmp排序

cpp 复制代码

vector<pair<int,int>> vec;
vec.emplace_back(0,2);
vec.emplace_back(1,5);
vec.emplace_back(1,9);
vec.emplace_back(4,6);
vec.emplace_back(5,9);
vec.emplace_back(8,10);

//默认按第一元素进行从小到大排序
sort(vec.begin(),vec.end());
//[0,2],[1,5],[1,9],[4,6],[5,9],[8,10]

1.外部定义

cpp 复制代码

static bool cmp(const pair<int, int> &a, const pair<int, int> &b) {
    if(b.first == a.first)
        return a.second < b.second;
    return a.first > b.first;
}

sort(vec.begin(),vec.end(),cmp);
//[8,10],[5,9],[4,6],[1,5],[1,9],[0,2]

2.匿名函数Lambda

cpp 复制代码

sort(vec.begin(),vec.end(),[&](const pair<int, int> &a, const pair<int, int> &b) {
    if(b.first == a.first)
        return a.second < b.second;
    return a.first > b.first;
});
//[8,10],[5,9],[4,6],[1,5],[1,9],[0,2]

这里也可以使用auto关键词，与上面用法一样