C++中std::set使用详解和综合示例

在 C++ 中，std::set 是一个标准容器，用于存储有序且唯一的元素。它基于**红黑树（Red-Black Tree）**实现，具有对元素的自动排序与快速查找、插入、删除等功能。

一、基本特性

二、常用操作与示例

1. 引入头文件

#include <set>
#include <iostream>

2. 基本用法：插入、遍历、查找

#include <set>
#include <iostream>

int main() {
    std::set<int> s;

    // 插入元素
    s.insert(5);
    s.insert(2);
    s.insert(8);
    s.insert(5); // 重复插入无效

    // 遍历元素（默认升序）
    std::cout << "Set elements: ";
    for (int x : s) {
        std::cout << x << " ";
    }
    std::cout << "\n";

    // 查找元素
    if (s.find(2) != s.end()) {
        std::cout << "Found 2 in the set.\n";
    }

    // 删除元素
    s.erase(8);

    // 判断是否为空
    std::cout << "Size: " << s.size() << "\n";
    std::cout << "Empty: " << (s.empty() ? "Yes" : "No") << "\n";

    return 0;
}

三、insert 返回值说明

auto [it, success] = s.insert(3);
if (success)
    std::cout << "3 inserted successfully.\n";
else
    std::cout << "3 already exists.\n";

四、自定义排序方式

示例：降序排列

struct Descend {
    bool operator()(int a, int b) const {
        return a > b; // 降序
    }
};

std::set<int, Descend> s = {5, 2, 8, 1};

五、常见成员函数

六、set 与 unordered_set 区别

七、使用场景举例

• 去重并排序数据（如字符串、整数集合）
• 动态维护最小/最大元素集合
• 实现滑动窗口、区间处理
• 作为图算法中的集合容器（如 Dijkstra、A*）

八、复杂示例：结构体去重与排序

#include <set>
#include <iostream>
#include <string>

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

    bool operator<(const Person& other) const {
        return age < other.age;  // 按年龄升序
    }
};

int main() {
    std::set<Person> people;
    people.insert({"Alice", 30});
    people.insert({"Bob", 25});
    people.insert({"Alice", 30}); // 被去重

    for (const auto& p : people) {
        std::cout << p.name << " - " << p.age << "\n";
    }
}

注意：set 的元素必须是可比较的（必须实现 operator< 或提供排序函数对象）。

九、其他综合示例

下面将详细讲解如何使用 std::set 实现常见的滑动窗口问题，比如 滑动窗口最大值、中位数 等。这类问题通常出现在面试题或算法竞赛中，利用 std::set 的有序性和对数时间复杂度非常合适。

示例 1：滑动窗口最大值（LeetCode 239）

问题描述：

给定一个整数数组 nums 和一个窗口大小 k，从左到右滑动窗口，找出每个窗口中的最大值。

Input: nums = [1,3,-1,-3,5,3,6,7], k = 3  
Output: [3,3,5,5,6,7]

std::multiset 解法（注意需要允许重复元素）

#include <vector>
#include <set>
#include <iostream>

std::vector<int> maxSlidingWindow(std::vector<int>& nums, int k) {
    std::multiset<int> window;
    std::vector<int> result;

    for (int i = 0; i < nums.size(); ++i) {
        window.insert(nums[i]);

        // 移除窗口左侧元素
        if (i >= k) {
            window.erase(window.find(nums[i - k]));
        }

        // 从第 k - 1 个位置开始收集结果
        if (i >= k - 1) {
            result.push_back(*window.rbegin()); // 最大值
        }
    }

    return result;
}

示例运行：

int main() {
    std::vector<int> nums = {1, 3, -1, -3, 5, 3, 6, 7};
    int k = 3;
    auto res = maxSlidingWindow(nums, k);
    for (int val : res) {
        std::cout << val << " ";
    }
    return 0;
}

输出：

3 3 5 5 6 7

示例 2：滑动窗口中位数（LeetCode 480）

解题思路：

将窗口拆分成：

• 左半部分（最大堆），维护较小的一半
• 右半部分（最小堆），维护较大的一半

用两个 std::multiset 近似维护这种结构。

C++ 代码（简化版）

#include <vector>
#include <set>
#include <iostream>

using namespace std;

class SlidingMedian {
    multiset<int> low, high;

    void rebalance() {
        while (low.size() > high.size()) {
            high.insert(*low.rbegin());
            low.erase(--low.end());
        }
        while (high.size() > low.size() + 1) {
            low.insert(*high.begin());
            high.erase(high.begin());
        }
    }

    void insert(int num) {
        if (high.empty() || num >= *high.begin()) high.insert(num);
        else low.insert(num);
        rebalance();
    }

    void erase(int num) {
        if (high.count(num)) high.erase(high.find(num));
        else low.erase(low.find(num));
        rebalance();
    }

    double getMedian() {
        if ((low.size() + high.size()) % 2 == 0)
            return ((double)*low.rbegin() + *high.begin()) / 2.0;
        else
            return *high.begin();
    }

public:
    vector<double> medianSlidingWindow(vector<int>& nums, int k) {
        vector<double> res;
        for (int i = 0; i < nums.size(); ++i) {
            insert(nums[i]);
            if (i >= k) erase(nums[i - k]);
            if (i >= k - 1) res.push_back(getMedian());
        }
        return res;
    }
};

示例运行：

int main() {
    vector<int> nums = {1, 3, -1, -3, 5, 3, 6, 7};
    int k = 3;
    SlidingMedian sm;
    auto res = sm.medianSlidingWindow(nums, k);
    for (auto r : res) {
        cout << r << " ";
    }
}

输出（中位数）：

1  -1  -1  3  5  6

总结：set 应用技巧