C++ 如何使用 set_C++ set 集合容器基础用法

C++中set是基于红黑树的有序唯一元素容器,支持O(log n)插入、查找、删除。需包含头文件,常用操作包括insert、erase、find、count等,元素不可修改,可自定义排序如std::greater实现降序。

在 C++ 中,set 是一种关联式容器,用于存储唯一元素,并自动按升序排序。它基于红黑树实现,查找、插入和删除操作的时间复杂度为 O(log n),适合需要有序且不重复数据的场景。

包含头文件与定义 set

使用 set 需要包含头文件 #include 。定义方式如下:

  • std::set s; —— 创建一个存放整数的集合
  • std::set<:string>words; —— 存放字符串的集合

常用操作方法

set 提供了多种成员函数来操作数据:

  • insert(value):插入元素,若已存在则不执行插入
  • erase(value):删除指定值的元素
  • find(value):查找元素,返回迭代器,未找到返回 end()
  • count(value):判断元素是否存在(返回 0 或 1)
  • size():返回元素个数
  • empty():判断是否为空
  • clear():清空所有元素

示例代码:

#include 
#include 
int main() {
    std::set s;
    s.insert(10);
    s.insert(5);
    s.insert(10); // 重复,不会插入
    s.insert(15);

    for (const auto& x : s) {
        std::cout << x << " "; // 输出:5 10 15
    }

    if (s.find(10) != s.end()) {
        std::cout << "\n找到了 10";
    }

    s.erase(5);
    std::cout << "\n大小:" << s.size(); // 输出:2
    return 0;
}

自定义排序规则

默认情况下,set 按升序排列。可以通过提供比较函数改变排序方式。例如降序:

std::set> desc_set;
desc_set.insert(10);
desc_set.insert(5);
desc_set.insert(15);
// 遍历输出:15 10 5

也可以传入自定义结构体或类,需重载比较运算符或提供仿函数。

注意事项

set 中的元素是只读的,修改会影响排序结构,因此不能直接修改元素值。如需更新,应先删除再插入新值。

基本上就这些。掌握 insert、find、erase 和遍历操作,就能应对大多数使用场景。不复杂但容易忽略细节,比如重复插入无效、遍历时不能修改等。用熟了很顺手。