C++ set容器如何保证唯一性 红黑树实现与自定义排序

std::set保证元素唯一性的核心机制在于其底层使用红黑树结构并结合排序规则。红黑树在插入时通过比较操作决定节点位置，若等于当前节点则不插入，从而避免重复；此外，红黑树的自平衡特性使操作复杂度稳定在O(log n)。自定义排序可通过提供比较函数改变排序逻辑，但必须满足严格弱序以确保正确判断唯一性。常见误区是若需存储重复值应使用std::multiset。实际应用中，存储自定义类型时需确保比较函数覆盖所有影响唯一性的字段。

C++ 中的 std::set 容器之所以能保证元素唯一性，核心机制在于其内部基于红黑树（Red-Black Tree）实现，并结合了排序规则。下面我们从底层结构和自定义排序两个角度来详细说明。

红黑树如何支持唯一性

std::set 底层使用的是红黑树这种自平衡二叉搜索树。红黑树在插入新节点时会根据比较结果决定插入位置，而这个比较过程直接决定了是否允许重复值存在。

• 插入新元素时，会与当前节点进行比较：
  • 如果小于当前节点，则进入左子树；
  • 如果大于当前节点，则进入右子树；
  • 如果等于当前节点，就不再插入。

换句话说，红黑树的查找逻辑天然地避免了重复值的插入，这就实现了 set 的唯一性。

此外，红黑树还保持了树的平衡性，使得插入、删除、查找的时间复杂度稳定在 O(log n)，这为 set 提供了高效的性能保障。

自定义排序影响唯一性的判断方式

默认情况下，std::set 使用 operator< 来比较元素大小。但如果你希望按照自己的规则排序，可以提供一个自定义的比较函数或仿函数。

例如：

struct MyCompare {
    bool operator()(const int& a, const int& b) const {
        return a > b; // 降序排列
    }
};
std::set<int, MyCompare> mySet;

在这个例子中，虽然排序顺序变了，但唯一性的判断仍然是通过比较函数来完成的。只要你的比较函数满足严格弱序（strict weak ordering），set 就能正常工作。

⚠️ 注意：如果你的比较函数没有正确处理相等关系，可能会导致 set 行为异常，比如误判重复或者漏判重复。

所以，自定义排序的关键点是：

• 保持比较逻辑的正确性；
• 不要遗漏对“等于”情况的处理（虽然不需要显式写等于，但要确保比较函数隐含这一逻辑）；
• 避免使用 operator==，因为 set 内部只用到了 < 或你提供的比较方式。

常见误区：为什么不能有重复值？

有些初学者可能会疑惑：“如果我想存多个相同值怎么办？”这时候应该考虑使用 std::multiset，它允许重复元素，同时仍然基于红黑树实现。

set 和 multiset 的区别就在于：

• set 在插入前检查是否存在相等的元素；
• multiset 直接插入，不管有没有相同的。

因此，如果你确实需要存储重复项，请选择 multiset。

实际应用中的小技巧

有时候我们想存储自定义类型对象到 set，这时候除了提供比较函数外，还需要注意几点：

• 类型必须可复制或可移动；
• 比较函数应覆盖所有可能影响唯一性的字段；
• 如果字段较多，建议重载 operator< 或使用 lambda（C++14+ 支持）；

举个例子：

struct Person {
    std::string name;
    int age;
};

struct ComparePerson {
    bool operator()(const Person& a, const Person& b) const {
        if (a.name != b.name) return a.name < b.name;
        return a.age < b.age;
    }
};

std::set<Person, ComparePerson> people;

这样设置后，只有当两个人的姓名和年龄都完全相同时，才会被视为重复，不会被插入。

基本上就这些。理解 set 的唯一性机制，关键在于搞清楚红黑树的插入逻辑和比较函数的作用方式。掌握了这两点，就能灵活运用并避免常见问题。