1.sort排序

排序

1.sort简介

sort函数包含在头文件<algorithm>中，在使用前需要引入头文件，有如下两种方式：

• #include <algorithm>
  

• #include <bits/stdc++.h> // 万能头文件
  

sort是C++标准库中的一个函数模板，用于对指定范围内的元素进行排序。

sort算法使用的是快速排序($QuickSort$)或者类似快速排序的改进算法，具有较好的平均时间复杂度，一般为$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; // > 代表降序, < 代表升序
});