C++ STL set如何保证唯一性？红黑树与自定义比较解析

C++ STL 中的 set 通过红黑树和比较函数保证元素唯一性。set 默认使用红黑树结构，其插入逻辑在查找过程中判断是否存在相等元素（由比较函数定义），若存在则不插入；用户可自定义比较函数以改变“相等”标准，但必须满足严格弱序要求；插入时返回 pair 可用于检测插入是否成功，且所有插入方式均遵循相同的唯一性判断机制。

C++ STL 中的 set 能够保证元素唯一性，主要依赖于其底层实现机制和插入时的判断逻辑。默认情况下，set 使用红黑树作为内部结构，并在插入元素时根据比较函数判断是否重复。理解这一点，有助于我们更好地使用 set，甚至自定义比较规则。

红黑树的结构特性决定了唯一性

set 的底层是基于红黑树实现的，这是一种自平衡的二叉查找树。红黑树的每个节点都满足一定的颜色和结构约束，使得树的高度始终保持在对数级别，从而保证了插入、查找和删除操作的时间复杂度为 O(log n)。

在 set 插入新元素时，红黑树会根据比较函数（默认是 less）来决定插入位置：

• 如果当前元素已经存在（即比较结果相等），则不会插入新的节点。
• 这个“相等”的判断标准，就是由比较函数决定的。

因此，红黑树本身的查找与插入逻辑，就天然支持了唯一性检查。

举个例子，假设你插入一个整数到 set：

std::set s;
s.insert(5);
s.insert(5); // 不会再次插入

第二次插入失败的原因，是在第一次插入后，红黑树中已存在值为 5 的节点，插入操作会在查找过程中发现“相等”，于是放弃插入。

自定义比较函数影响唯一性判断

默认情况下，set 使用的是升序排列（即 std::less），这意味着它通过 operator< 来判断两个元素是否“相等” —— 如果 a < b 和 b < a 都不成立，则认为它们相等。

但如果我们想改变这个判断方式，比如让 set 按照降序排序，或者处理自定义类型时，就需要提供自己的比较函数或函数对象。例如：

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

在这个例子中，set 依然能保证唯一性，只是判断依据变了：现在是根据“大于”来判断顺序，进而判断是否相等。

对于自定义类来说，比如：

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

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

只要你的比较函数符合“严格弱序”（strict weak ordering）的要求，就能正确支持 set 的唯一性和排序功能。

注意：如果你的比较函数没有正确实现，可能会导致：

• 插入重复元素
• 查找失败
• 树结构混乱，甚至程序崩溃

插入行为中的细节处理

当我们调用 insert() 方法时，返回值其实是一个 pair，其中第二个元素表示插入是否成功。这可以用来判断是否发生了重复插入：

auto result = s.insert(5);
if (!result.second) {
    std::cout << "元素已存在，插入失败\n";
}

此外，set 还提供了其他插入方式，比如使用初始化列表、范围插入等，但底层逻辑一致：每次插入都会进行查找，若找到相等元素则放弃插入。

另一个容易忽略的点是：即使两个对象的内容完全相同，但如果比较函数未正确定义，也可能被视为不同元素。所以在使用自定义类型时，务必确保比较函数能够覆盖所有关键字段。

基本上就这些。掌握红黑树的基本原理和比较函数的作用机制，才能真正理解 set 是如何做到元素唯一的。