1.前言
在日常大家写代码时,或者刷算法题中,如果要对数组里面的数进行一个排序时,我们常常会写一个快速排序,或者是冒泡排序,但也许我们使用stl库中的sort函数,能减少不少代码量以及不容易出错。
2.介绍
STL(标准模板库)中的sort
函数是用于对序列进行排序的算法之一。它的作用是将一个序列中的元素按照指定的顺序重新排列。
3.底层代码
通常来说,sort
函数的底层代码涉及到快速排序算法的一些变种,例如经典的快速排序或者改进的版本(比如introsort,它在最坏情况下会退化为堆排序以保证性能)。以下是一个sort
函数的底层实现:
template<typename RandomAccessIterator, typename Compare>
void sort(RandomAccessIterator first, RandomAccessIterator last, Compare comp) {
if (first == last || std::next(first) == last)
return; // 如果范围为空或只有一个元素,则无需排序
auto pivot = *std::next(first, std::distance(first, last) / 2); // 选择中间元素作为枢轴
// 分割范围为两部分,小于枢轴的放左边,大于枢轴的放右边
auto middle1 = std::partition(first, last, [=](const auto& elem) { return comp(elem, pivot); });
auto middle2 = std::partition(middle1, last, [=](const auto& elem) { return !comp(pivot, elem); });
// 递归排序左右两部分
sort(first, middle1, comp);
sort(middle2, last, comp);
}
其中使用了STL中的partition
函数来实现元素的分割,然后递归地对左右两部分进行排序。这是一个简化的实现,实际的STL实现可能会更加复杂,例如可能包括对小范围使用插入排序来提高性能等优化。同时,实际的STL实现还可能包括对不同数据类型的优化和特化。
4.使用
4.1
头文件:
#include<stdio.h>
函数:
sort(begin,end,cmp)
begin指向数组的第一个指针,end指向数组最后元素的下一个指针,cmp可以自己定义,也可以不写,如果不写的话会默认将数组从小到大排列。
如果是从大到小的话,得写成sort(a,a+10,greater<int>());
此外,cmp也可以自己定义,以此按照编码者意愿实现响应的功能
例如,如果要对一个存储自定义结构体的容器按照结构体中某个成员变量进行排序,你可以编写一个自定义的比较函数,然后将其传递给sort
函数。
#include <algorithm>
#include <vector>
struct Person {
std::string name;
int age;
};
// 比较函数,按照年龄进行排序
bool compareByAge(const Person& a, const Person& b) {
return a.age < b.age;
}
int main() {
std::vector<Person> people = {{"Alice", 30}, {"Bob", 25}, {"Charlie", 35}};
// 按照年龄进行升序排序
std::sort(people.begin(), people.end(), compareByAge);
return 0;
}
这段代码中,sort
函数将按照compareByAge
函数所定义的规则对people
容器中的元素进行排序,即按照年龄的升序排列。