排序
1.sort简介
sort函数包含在头文件<algorithm>中,在使用前需要引入头文件,有如下两种方式:
-
#include <algorithm>
-
#include <bits/stdc++.h> // 万能头文件
sort是C++标准库中的一个函数模板,用于对指定范围内的元素进行排序。
sort算法使用的是快速排序( Q u i c k S o r t QuickSort QuickSort)或者类似快速排序的改进算法,具有较好的平均时间复杂度,一般为 O ( n l o g n ) O(nlogn) O(nlogn)。
2.sort用法
sort(起始地址,结束地址的下一位,*比较函数)
*号的意思代表比较函数参数是可选的,默认是升序,从小到大
2.1 对数组元素进行排序
案例一:下面对数组进行分别进行升序和降序排序
#include <bits/stdc++.h>
using namespace std;
// 降序的比较函数
bool cmp_de(int u, int v) {
return u > v;
}
// 升序的比较函数 --- 一般不写, 默认为升序
bool cmp_up (int u, int v) {
return u < v;
}
int main() {
int n;
scanf("%d", &n);
int a[n];
// 读取
for(int i = 0; i < n ; i ++) {
scanf("%d", &a[i]);
}
// 降序
sort(a, a + n, cmp_de); // a等价于&a[0], a + n等价于a[n-1]的下一个地址,不等价&a[n],不能代替a + n,因为产生了越界,所以推荐上述的写法为a + n。
// 降序输出
for(int i = 0; i < n; i ++) {
printf("%d ", a[i]);
}
printf("\n");
//升序
sort(a, a + n);// sort(a, a + n, cmp_up);
//升序输出
for(int i = 0; i < n; i ++) {
printf("%d ", a[i]);
}
return 0;
}
案例二:使用默认的升序排序
#include <bits/stdc++.h>
using namespace std;
int main() {
int a[1000];
int n;
// 读取数组大小
cin >> n;
// 读取元素
for(int i = 1; i <= n; ++ i) cin >> a[i];
// 对数组进行排序, 默认升序排序
sort(a + 1, a + n + 1); // a + 1对应的a[1]的地址, a + n + 1对应的是a[n+1]的地址, 也就是a[n]的下一个地址
for(int i = 1; i <= n; i ++) {
cout << a[i] << " ";
}
return 0;
}
输入
5
2 3 6 9 1
输出
1 2 3 6 9
2.2 STL排序
案例一:对于vector进行排序
#include <bits/stdc++.h>
using namespace std;
int main() {
// 初始化v
vector<int> v = {5, 1, 3, 9, 11};
// 对vector数组进行排序
sort(v.begin(), v.end());
// v.begin() 返回指向向量 v 中第一个元素的迭代器。
// v.end() 返回指向向量 v 中最后一个元素的下一个位置的迭代器。
for(int i = 0; i < v.size(); i ++ ) {
cout << v[i] << " ";
}
// 输出结果:1 3 5 9 11
return 0;
}
如果需要传入cmp比较函数,参考2.1的cmp写法,是一致的。
2.3 struct排序
2.3.1 符号重载支持sort排序
sort函数无法直接对struct进行排序,但是对于struct的<或>进行重载后就可以排序了。
- 重载了<号,sort默认可以升序排序,或者比较参数为less(),代表使用重载 < 运算符
- 重载了>号,sort可以降序排序,比较函数参数为greater(),代表使用重载 > 运算符
#include <bits/stdc++.h>
using namespace std;
struct Node {
int u, v;
// 重载 < 运算符
bool operator < (const Node &m) const {
return u == m.u ? v < m.v : u < m.u;
}
// 重载 > 运算符
bool operator > (const Node &m) const {
return u == m.u ? v > m.v : u > m.u;
}
};
int main() {
Node a = {1, 2};
Node b = {1, 3};
Node c = {1, 4};
Node d[3] = {a, c, b};
// 默认升序,升序输出
sort(d, d + 3);// sort(d, d + 3, less<Node>());
for(int i = 0; i < 3; i ++) {
cout << d[i].u << " " << d[i].v << endl;
}
cout << "\n";
//降序,降序输出
sort(d, d + 3, greater<Node>());
for(int i = 0; i < 3; i ++) {
cout << d[i].u << " " << d[i].v << endl;
}
return 0;
}
/*
1 2
1 3
1 4
1 4
1 3
1 2
*/
2.3.2 改写sort的比较函数
sort函数无法直接对struct进行排序,通过改写比较函数可以完成升降序排序
#include <bits/stdc++.h>
using namespace std;
struct Node {
int u, v;
};
// 比较函数
bool compareNodes_de(const Node &a, const Node &b) {
return a.u == b.u ? a.v > b.v : a.u > b.u;
}
// 比较函数
bool compareNodes_up(const Node &a, const Node &b) {
return a.u == b.u ? a.v < b.v : a.u < b.u;
}
int main() {
Node a = {1, 2};
Node b = {1, 3};
Node c = {1, 4};
Node d[3] = {a, c, b};
// 降序排序且输出
sort(d, d + 3, compareNodes_de);
for(int i = 0; i < 3; i ++) {
cout << d[i].u << " " << d[i].v << endl;
}
cout << "\n";
// 升序排序且输出
sort(d, d + 3, compareNodes_up);
for(int i = 0; i < 3; i ++) {
cout << d[i].u << " " << d[i].v << endl;
}
return 0;
}
/*
1 4
1 3
1 2
1 2
1 3
1 4
*/
3. 自定义比较函数
sort默认使用小于号(升序)进行排序,如果想要自定义比较规则,可以传入第三个参数,可以是函数或者lambda表达式(匿名函数)。
降序排序-函数形式
bool cmp(const int &u, const int &v) { // 参数可为int u, int v
return u > v; // > 代表降序, < 代表升序
}
...
sort(..., ..., cmp);
降序排序-lambda表达式(匿名函数)
sort(..., ..., [](const int &u, const int &v){// 参数可为int u, int v
return u > v; // > 代表降序, < 代表升序
});