使用C++ sort函数对vector进行自定义排序

2023-07-26
关注

>>> 背景

今天在学一些C++ STL容器,看到sort函数允许自定义排序规则,小小地实操了一下。

>>> 内容

vector

在正式开始使用sort之前,我们先铺垫一些关于vector容器的内容,以及自定义标准输出流,为我们后面的探索铺垫一下。

vector容器是一种动长的模板容器,需要 # include 。定义一个vector对象需要指定元素类型,这个类型可以是基础类型,结构体,自定义类型和vector类型自身(达到嵌套多维vector的目的)。

如下面代码所示,创建了一个元素类型为 int 的vector对象,并使用列表进行初始化。除此之外,还可以使用assign方法进行快速初始化。

// 元素为 1,2,3,4,5
vector< int > vec(5) = {1, 2, 3, 4, 5}; 
// 元素为 1,1,1,1,1
vector< int > vec(5);
vec.assign(1, 5)

还可以创建二维容器,如下所示。注意到此时的模板元素类型参数变成了 vector< int >。

vector< vector< int >> vec2 = {vec, vec, vec, vec, vec};

为了能够方便地打印出vector对象的元素,需要自定义一些输出函数。这里分别对一维和二维的vector容器定义了打印函数(模板)。

template < typename T >
void print1vec(const vector< T >& v) {
    auto iter = v.begin();
    while (iter != v.end()) {
        cout < < *iter < < " ";
        iter++;
    }
    cout < < endl;
}


template < typename T >
void print2vec(const vector< vector< T >>& v) {
    auto iter = v.begin();
    while (iter != v.end()) {
        print1vec(*iter);
        iter++;
    }
    cout < < endl;
}

为了让输出更加优雅和方便,可以重载<<运算符,如下所示。

template < typename T >
ostream& operator< < (ostream& os, const vector< T >& v) {
    auto iter = v.begin();
    while (iter != v.end()) {
        os < < *iter < < " ";
        iter++;
    }
    os < < endl;
    return os;
};


template < typename T >
ostream& operator< < (ostream& os, const vector< vector< T >>& v) {
    auto iter = v.begin();
    while (iter != v.end()) {
        os < < *iter;
        iter++;
    }
    os < < endl;
    return os;
};

sort

使用sort需要 #include 。sort函数最简单的用法如下。第一个参数传入容器的头地址(或指向头元素的迭代器),第二个参数传入容器的尾地址(或指向尾元素的迭代器)的下一个位置(前闭后开的区间)。

int arr[5] = {1, 3, 5, 2, 4};
    sort(arr, arr + 5);
    for (int i = 0; i < 5; ++i)
        cout < < arr[i] < < " ";
>> 1 2 3 4 5

还可以传入第3个参数,即排序判据。该判据由一个函数对象来实现,有两个参数,返回bool值。sort在排序时,会依次把两个元素传入给该函数对象进行判断,如果返回值为true,那么就会把第一个参数代表的元素放到队列中靠前的位置。

因此,可以传入判据函数来快速实现降序排序(这里使用了lambda表达式)。如下所示。

int arr[5] = {1, 3, 5, 2, 4};
sort(arr, arr + 5,
        [](int a, int b) {
            return a > b;
        });
    for (int i = 0; i < 5; ++i)
        cout < < arr[i] < < " ";

5 4 3 2 1

判据函数给了用户自己决定元素比较关系的途径,尤其是对一些非基础类型的数据(如结构、类)可以方便地进行排序。

下面就创建了一个结构体 ,并规定如果该结构体的两个成员的平方和越小,排序越靠前。

struct Point {
    int x; 
    int y;
};


ostream& operator< < (ostream& os, const Point& p) {
    os < < "(" < < p.x < < ", " < < p.y < < ")";
    return os;
};
vector< Point > vec3;
Point p1 = {1, 1};
Point p2 = {-3, -3};
Point p3 = {2, 2};
vec3.push_back(p1);
vec3.push_back(p2);
vec3.push_back(p3);
cout < < "vec3: " < < endl;
cout < < vec3;
sort(vec3.begin(), vec3.end(),
    [](Point a, Point b){
        return a.x * a.x +  a.y * a.y < b.x * b.x +  b.y * b.y;
    });
cout < < vec3;
> > vec3:
(1, 1) (-3, -3) (2, 2)
(1, 1) (2, 2) (-3, -3)

  • “人造太阳”计划
  • vector
  • sort
  • vector初始化
您觉得本篇内容如何
评分

评论

您需要登录才可以回复|注册

提交评论

慧生活

这家伙很懒,什么描述也没留下

关注

点击进入下一篇

图解transformer中的自注意力机制

提取码
复制提取码
点击跳转至百度网盘