STL里面的sort或自定义排序

renshy5

已于 2023-04-05 17:18:10 修改

阅读量581

点赞数

分类专栏：写代码啦文章标签： c++ 算法数据结构

于 2023-04-05 17:07:42 首次发布

本文链接：https://blog.csdn.net/renshy5/article/details/129973379

版权

写代码啦专栏收录该内容

12 篇文章 0 订阅

订阅专栏

1. vector

1.1 默认排序函数（升序）

vector里面有sort，使用<algorithm>库文件即可，使用方法如下：

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int main()
{
    vector<int> nums = {1, 8, 4, 2, 12};
    sort(nums.begin(), nums.end());
    for (int i : nums)
        cout << i << " ";
    // 1 2 4 8 12
    return 0;
}

也可以对char、string、pair默认升序排序，其他double之类的默认类型也是可以的，不予重述。

char：直接根据ASCII来升序排序的

vector<char> my_char = {'z', 'c', 's', 'e', 'x'};
sort(my_char.begin(), my_char.end());
for (char i : my_char)
    cout << i << " ";
// c e s x z

string：根据字典序排序的

vector<string> my_string = {"boy", "bag", "banana", "apple", "app", "zoo"};
sort(my_string.begin(), my_string.end());
for (string i : my_string)
    cout << i << " ";
// app apple bag banana boy zoo

pair：会首先按照pair中的第一个key排序，然后key相同，对value排序

vector<pair<int, int>> my_pair = {{1, 12}, {2, 3}, {1, 2}};
sort(my_pair.begin(), my_pair.end());
for (pair<int, int> i : my_pair)
    cout << i.first << " " << i.second << endl;
// 1 2
// 1 12
// 2 3

1.2 自定义排序

有时候需要自定义类型的比较，或者按照自己的比较方式来排序，这时候就需要自定义排序了

// 自定义比较函数，这里是按照一个pair的value大小来排序
bool cmp(pair<string, int> a, pair<string, int> b)
{
    return a.second < b.second;
}

int main()
{
    vector<pair<string, int>> my_pair = {{"abs", 2}, {"tmp", 31}, {"ceil", 1}};
    sort(my_pair.begin(), my_pair.end(), cmp);
    for (pair<string, int> i : my_pair)
        cout << i.first << " " << i.second << endl;
    // ceil 1
    // abs 2
    // tmp 31
    return 0;
}

2. map

2.1 默认排序（升序）

默认的map其实是排序的map，它底层的实现是一个红黑树，所以当插入数据的时候会默认根据key来排序。

map<int, int> my_map = {{3, 12}, {2, 3}, {1, 2}};
for (auto i : my_map)
    cout << i.first << " " << i.second << endl;
// 1 2
// 2 3
// 3 12

类似的同上面的vector的一样。

2.2 根据`key`来自定义排序

这里要想自定义类型或者自定义比较方式来排序，分为两种，一种是根据key来自定义类型或比较方式排序；另一种是根据value来自定义类型或比较方式排序。两种不一样，所以需要分开介绍。这里先说根据key来自定义排序

// 比较器，这不是比较函数了
struct cmp
{
    bool operator()(string a, string b)
    {
        if (a.size() == b.size())
            return a < b;
        return a.size() < b.size();
    }
};
int main()
{
    map<string, int, cmp> my_map = {{"a", 100}, {"tmp", 31}, {"zoo", 4}, {"abs", 2}, {"ceil", 1}};
    for (auto i : my_map)
        cout << i.first << " " << i.second << endl;
    // a 100
    // abs 2
    // tmp 31
    // zoo 4
    // ceil 1
    return 0;
}

2.3 根据`value`来自定义排序

根据value来排序就比较麻烦了，一般有两个解决方案：

再建立一个new_map，把原来的my_map的key和value换一下插入到new_map里面，然后根据key来排序。
再建一个vector，把原来的my_map插入vector里面，然后排序。

以上两种方案都可以在上面找到，这里不重复叙述了。

3. set

3.1 默认排序（升序）

set默认是排序的，和map一样，底层也是红黑树，所以自带排序（升序）

set<string> my_set = {"boy", "bag", "banana", "apple", "app", "zoo"};
for (auto i : my_set)
    cout << i << " ";
// app apple bag banana boy zoo

3.2 自定义排序

和map的按key排序一样，自己写一个比较器就可以了，不予重述。

4. priority_queue

4.1 默认排序（大顶堆）

priority_queue的底层实现的数据结构是堆，默认是大顶堆，就是最大的始终在顶上。（虽然底层的数据结构是堆，但是因为建立的是完全二叉树，所以其实容器是vector）

priority_queue<int> my_pq;
vector<int> nums = {1, 8, 4, 2, 12};
for (int i : nums)
    my_pq.push(i);
while (!my_pq.empty())
{
    cout << my_pq.top() << " ";
    my_pq.pop();
}
// 12 8 4 2 1

4.2 自定义排序

前面说了priority_queue底层的容器是vector，所以在建堆的时候要指定容器。

// 自定义比较器
struct cmp
{
    bool operator()(string a, string b)
    {
        if (a.size() == b.size())
            return a < b;
        return a.size() < b.size();
    }
};
int main()
{
  // 指定容器是vector<string>，比较器是cmp
    priority_queue<string, vector<string>, cmp> my_pq;
    vector<string> nums = {"boy", "bag", "banana", "apple", "app", "zoo"};
    for (string i : nums)
        my_pq.push(i);
    while (!my_pq.empty())
    {
        cout << my_pq.top() << " ";
        my_pq.pop();
    }
    // banana apple zoo boy bag app
    return 0;
}