c++如何自定义比较函数

---

前言

在 C++ 编程中，自定义比较函数是一种重要的技术，它使得开发者能够根据特定标准对数据进行排序和比较。这一技术在处理标准模板库（STL）容器如 std::sort, std::set, std::map, std::priority_queue 等时尤为关键。能够定义自定义比较逻辑不仅提升了代码的灵活性和效率，还能解决许多复杂的问题。本文将探讨在 C++ 中如何创建和使用自定义比较函数，包括使用函数对象（Functors）、Lambda 表达式、标准库比较函数、普通函数，以及重载 < 运算符的方法。每种方法都适用于特定的场景，使得数据排序和组织更加高效和直观。

---

在普通函数中使用

1. 使用函数对象（Functor）
   函数对象是一个重载了 operator() 的类的实例。这种方法非常灵活，因为您可以在类内部存储额外的状态。

要特别注意，在使用sort等数据结构时需要加括号，也就是需要传入实例对象来使用，而优先队列priority_queue则不用。

#include <queue>
#include <vector>

class Compare {
public:
    bool operator()(int a, int b) {
        // 返回 true 如果 a 应该排在 b 之后
        return a > b;  // 这将创建一个最小堆
    }
};

int main() {
    std::priority_queue<int, std::vector<int>, Compare> pq; // 传入的是类
    // 使用 pq...
    vector<int> arr; // 定义一个数组
    // 初始化arr ...
    std::sort(arr.begin(), arr.end(), Compare()); // 传入的是对象
}

2. 使用 Lambda 函数
   在 C++11 及更高版本中，您可以使用 lambda 函数作为比较函数。这种方法更加简洁和直观。

#include <queue>
#include <vector>

int main() {
    auto compare = [](int a, int b) { return a > b; };  // 创建最小堆
    std::priority_queue<int, std::vector<int>, decltype(compare)> pq(compare);
    // 使用 pq...
}

3. 使用标准库函数对象
   您还可以使用标准库中的预定义函数对象，如 std::greater 或 std::less。这是一种简单快捷的方法，特别是当您需要基本的比较逻辑时。

#include <queue>
#include <functional>

int main() {
    std::priority_queue<int, std::vector<int>, std::greater<int>> pq;
    // 使用 pq...
}

4. 使用自定义函数
   对于非 lambda 表达式和非类类型的函数对象，您可以使用函数指针。

#include <queue>
#include <vector>

bool compare(int a, int b) {
    return a > b;  // 创建最小堆
}

int main() {
    std::priority_queue<int, std::vector<int>, bool(*)(int, int)> pq(compare);
    // 使用 pq...
}

---

在类中使用

在自定义类中使用自定义比较函数可以让类更加灵活，并且能够适应更多样化的排序或比较需求。例如，假设有一个表示学生的类 Student，想根据学生的成绩或姓名来进行排序。以下是如何实现这一点的示例：

1. 示例：自定义类 Student 和其比较函数

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

class Student {
public:
    std::string name;
    int score;

    Student(std::string name, int score) : name(name), score(score) {}
};
// 按分数比较
bool compareByScore(const Student &a, const Student &b) {
    return a.score > b.score; // 降序
}

// 按姓名比较
bool compareByName(const Student &a, const Student &b) {
    return a.name < b.name; // 升序
}

在 main 函数中使用这些比较函数：

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

    // 按分数排序
    std::sort(students.begin(), students.end(), compareByScore);
    std::cout << "Sorted by score:" << std::endl;
    for (const auto &student : students) {
        std::cout << student.name << ": " << student.score << std::endl;
    }

    // 按姓名排序
    std::sort(students.begin(), students.end(), compareByName);
    std::cout << "\nSorted by name:" << std::endl;
    for (const auto &student : students) {
        std::cout << student.name << ": " << student.score << std::endl;
    }

    return 0;
}

2. 示例：重载 < 运算符

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

class Student {
public:
    std::string name;
    int score;

    Student(std::string name, int score) : name(name), score(score) {}

    // 重载 < 运算符
    bool operator < (const Student& other) const {
        return this->score < other.score;  // 根据成绩升序排列
    }
};

在这个示例中，< 运算符被重载为根据学生的成绩进行比较，可以直接使用 std::sort 对 Student 对象的数组进行排序，而无需指定自定义比较函数：

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

    // 使用重载的 < 运算符排序
    std::sort(students.begin(), students.end());
    std::cout << "Sorted by score:" << std::endl;
    for (const auto& student : students) {
        std::cout << student.name << ": " << student.score << std::endl;
    }

    return 0;
}

---

总结

本文详细探讨了在 C++ 中创建和使用自定义比较函数的多种方法。从函数对象、Lambda 表达式到标准库比较函数，每种方法都有其适用场景。特别地，通过重载 < 运算符，可以使自定义类直接支持标准库容器的默认比较机制，这在实际编程中非常实用。理解这些不同的实现方法对于提高代码的灵活性和解决复杂问题至关重要。根据不同的需求和场景选择合适的比较逻辑，可以大大增强程序的功能性和可维护性。