再学C++ | std::set 的原理

std::set 是C++标准库中的容器之一,它基于红黑树实现。std::set 利用红黑树的特性来实现有序的插入、查找和删除操作,并且具有较好的平均和最坏情况下的时间复杂度。
当向 std::set 插入元素时,它会按照特定的比较函数(bool less::operator() const(const T &lhs, const T & rhs))将新元素插入到红黑树的适当位置,以保持树的有序性质。插入操作的平均时间复杂度为 O ( log ⁡ n ) O(\log n) O(logn),其中 n n nstd::set 中元素的数量。查找操作(find())使用红黑树的性质,通过比较函数在树中进行二分查找,查找操作的平均时间复杂度为 O ( log ⁡ n ) O(\log n) O(logn)

但是当我们把 struct 放入 std::set 会有什么后果呢?因为 std::set 需要在插入到时候排序,所以需要重载 struct 的比较运算符,这个时候就出现问题了,首先我们定义一个结构体 Person

struct Person {
    Person(int _ID, string _name, int _age) : ID(_ID), age(_age), name(_name) {}
    int ID;
    int age;
    string name;
};

当我们直接插入到 std::set 中时,会报 complier error 的错误,因此简单补写一个比较运算符重载,如下:

bool operator<(const Person &lhs, const Person &rhs) {
    return lhs.age < rhs.age;
}

OK,编译起来没有问题,但是我们运行测试一下下面的find操作就会发现问题

#include 
#include 

using namespace std;

struct Person {
    Person(int _ID, string _name, int _age) : ID(_ID), age(_age), name(_name) {}
    int ID;
    int age;
    string name;
};

bool operator<(const Person &lhs, const Person &rhs) {
    return lhs.age < rhs.age;
}

int main() {
    set<Person> person;
    for (int i = 0; i < 1000; ++i) {
        Person p_tmp(i, "sxj", 10);
        person.insert(p_tmp);
    }
    Person p(2000, "sxj", 10);
    auto it = person.find(p);
    if (it != person.end())
        cout << "Find Person --- ID: " << it->ID << "  name: " << it->name << "  age: " << it->age;
    else
        cout << "Can't find" << endl;
    return 0;
}

运行结果

明明不在set中的 ID-2000Person也可以被找到。造成这个结果的原因是我们所提供的 operator<() ,当Person p1、p2,在 p1find 就会判断 p1 和 p2 是同一个 Person ,因此会造成这样的错误结果。

解决方案就是补充完整我们的比较运算符重载,完整代码如下

#include 
#include 

using namespace std;

struct Person {
    Person(int _ID, string _name, int _age) : ID(_ID), age(_age), name(_name) {}
    int ID;
    int age;
    string name;
};

bool operator<(const Person &lhs, const Person &rhs) {
    if (lhs.ID < rhs.ID) return true;
    if (lhs.ID > rhs.ID) return false;
    if (lhs.name < rhs.name) return true;
    if (lhs.name > rhs.name) return false;
    return lhs.age < rhs.age;
}

int main() {
    set<Person> person;
    for (int i = 0; i < 1000; ++i) {
        Person p_tmp(i, "sxj", 10);
        person.insert(p_tmp);
    }
    Person p(2000, "sxj", 10);
    auto it = person.find(p);
    if (it != person.end())
        cout << "Find Person --- ID: " << it->ID << "  name: " << it->name << "  age: " << it->age;
    else
        cout << "Can't find" << endl;
    return 0;
}

你可能感兴趣的:(#,C/C++,c++,开发语言)