商城首页欢迎来到中国正版软件门户

您的位置:首页 >C++中求两个set的差集方法

C++中求两个set的差集方法

  发布于2025-10-04 阅读(0)

扫一扫,手机访问

使用std::set_difference可求两个set的差集,需包含<algorithm>头文件,输入set自动有序,输出容器需预分配空间或使用inserter;示例中set1与set2的差集为{1,2},结果存入vector或set;自定义类型需重载<运算符以支持比较。

c++中如何求两个set的差集_c++ set差集方法

在C++中,求两个set的差集可以使用标准库中的std::set_difference算法。这个函数定义在algorithm头文件中,用于计算两个有序序列的差集,即属于第一个集合但不属于第二个集合的元素。

使用 set_difference 求差集

步骤说明:

  • std::set本身是有序容器,天然满足set_difference对有序输入的要求。
  • 调用std::set_difference时,需要提供两个输入区间的迭代器和一个输出迭代器。
  • 结果会写入到指定的输出容器中,比如另一个setvector

示例代码:

#include <iostream>
#include <set>
#include <vector>
#include <algorithm>
int main() {
    std::set<int> set1 = {1, 2, 3, 4, 5};
    std::set<int> set2 = {3, 4, 5, 6, 7};

    std::vector<int> difference;
    // 预分配空间,避免多次扩容
    difference.resize(set1.size());

    auto it = std::set_difference(
        set1.begin(), set1.end(),
        set2.begin(), set2.end(),
        difference.begin()
    );

    // 调整大小以去除未使用的部分
    difference.erase(it, difference.end());

    std::cout << "差集: ";
    for (const auto& elem : difference) {
        std::cout << elem << " ";
    }
    std::cout << std::endl;

    return 0;
}

输出结果:
差集: 1 2

注意事项

  • std::set_difference要求输入区间已排序,而std::set自动有序,因此可直接使用。
  • 输出容器需提前分配足够空间,否则可能导致未定义行为。
  • 若希望结果仍为set类型,可将结果插入新的std::set,或直接用inserter避免手动管理空间。

更简洁的写法(使用 inserter):

std::set<int> result;
std::set_difference(
    set1.begin(), set1.end(),
    set2.begin(), set2.end(),
    std::inserter(result, result.begin())
);

自定义类型的差集

如果set中存储的是自定义类型,需确保该类型支持比较操作(即定义了<运算符),或者传入相应的比较函数。

例如:

struct Person {
    int id;
    std::string name;
};
bool operator<(const Person& a, const Person& b) {
    return a.id < b.id;
}

只要重载了<,就可以像基本类型一样使用set_difference

基本上就这些。只要包含头文件、正确使用迭代器和输出容器,就能高效求出两个set的差集。
本文转载于:互联网 如有侵犯,请联系zhengruancom@outlook.com删除。
免责声明:正软商城发布此文仅为传递信息,不代表正软商城认同其观点或证实其描述。

热门关注