C++ sort()函数

介绍

sort() 函数用于对容器中的元素进行排序。默认情况下，它使用升序排序。可以通过传递自定义的比较函数来实现自定义排序。

语法

template <class RandomIt>
void sort(RandomIt first, RandomIt last);

template <class RandomIt, class Compare>
void sort(RandomIt first, RandomIt last, Compare comp);

• first：排序范围的起始迭代器。
• last：排序范围的结束迭代器。
• comp：自定义比较函数，可选。

示例代码

#include <iostream>
#include <vector>
#include <algorithm>

int main() {
    std::vector<int> vec = {5, 2, 9, 1, 5, 6};

    std::sort(vec.begin(), vec.end()); // 默认升序排序
    std::cout << "Sorted in ascending order: ";
    for (int num : vec) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    std::sort(vec.begin(), vec.end(), std::greater<int>()); // 使用自定义比较函数降序排序
    std::cout << "Sorted in descending order: ";
    for (int num : vec) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    return 0;
}

在 C++ 中，sort() 函数可以通过传递一个自定义的比较函数或函数对象来实现自定义排序规则。这个比较函数决定了排序的顺序。比较函数必须接受两个参数并返回一个布尔值，表示第一个参数是否应在第二个参数之前。

自定义排序规则

1. 使用函数指针

你可以定义一个普通的函数作为比较函数。该函数返回 true 表示第一个参数应在第二个参数之前，否则返回 false。

#include <iostream>
#include <vector>
#include <algorithm>

bool customCompare(int a, int b) {
    // 定义升序排序规则
    return a < b;
}

int main() {
    std::vector<int> vec = {5, 2, 9, 1, 5, 6};

    std::sort(vec.begin(), vec.end(), customCompare);
    std::cout << "Sorted vector (ascending order): ";
    for (int num : vec) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    return 0;
}

2. 使用函数对象（仿函数）

可以通过定义一个类并重载 operator() 操作符来创建函数对象。这样可以更灵活地保存状态和数据。

#include <iostream>
#include <vector>
#include <algorithm>

struct CustomCompare {
    bool operator()(int a, int b) const {
        // 定义降序排序规则
        return a > b;
    }
};

int main() {
    std::vector<int> vec = {5, 2, 9, 1, 5, 6};

    std::sort(vec.begin(), vec.end(), CustomCompare());
    std::cout << "Sorted vector (descending order): ";
    for (int num : vec) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    return 0;
}

3. 使用 Lambda 表达式

Lambda 表达式是定义比较函数的一种简洁方式，尤其适用于简单的排序规则。

#include <iostream>
#include <vector>
#include <algorithm>

int main() {
    std::vector<int> vec = {5, 2, 9, 1, 5, 6};

    std::sort(vec.begin(), vec.end(), [](int a, int b) {
        // 定义升序排序规则
        return a < b;
    });

    std::cout << "Sorted vector (ascending order): ";
    for (int num : vec) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    return 0;
}

自定义排序规则的应用

按结构体成员排序

在实际应用中，我们经常需要根据对象的某个成员变量进行排序。例如，按学生的成绩进行排序：

#include <iostream>
#include <vector>
#include <algorithm>

struct Student {
    std::string name;
    int score;
};

// 自定义比较函数，按成绩降序排序
bool compareByScore(const Student& a, const Student& b) {
    return a.score > b.score;
}

int main() {
    std::vector<Student> students = {
        {"Alice", 90},
        {"Bob", 85},
        {"Charlie", 95},
        {"Dave", 80}
    };

    std::sort(students.begin(), students.end(), compareByScore);

    std::cout << "Students sorted by score (descending order):" << std::endl;
    for (const Student& student : students) {
        std::cout << student.name << ": " << student.score << std::endl;
    }

    return 0;
}