c++如何求两个vector的交集和差集 _c++ vector交集与差集计算方法

使用set_intersection和set_difference前需排序两vector,分别求交集与差集,结果存入预分配或通过back_inserter动态扩展的容器中。

在C++中,求两个vector的交集和差集可以通过标准库中的算法函数实现,前提是这两个vector已经排序。主要使用头文件中的set_intersectionset_difference函数。

1. 求两个vector的交集(set_intersection)

要计算两个vector的交集,需先对两个vector进行排序,然后使用set_intersection将结果写入目标容器。

示例代码:

#include iostream>
#include
#include gorithm>
using namespace std;

int main() {
    vector v1 = {1, 2, 5, 6};
    vector v2 = {2, 3, 4, 5};
    vector v_intersection;

    // 必须先排序
    sort(v1.begin(), v1.end());
    sort(v2.begin(), v2.end());

    // 预分配空间
    v_intersection.resize(min(v1.size(), v2.size()));

    auto it = set_intersection(
        v1.begin(), v1.end(),
        v2.begin(), v2.end(),
        v_intersection.begin()
    );

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

    cout     for (int n : v_intersection) cout     cout
    return 0;
}

输出:
交集: 2 5

2. 求两个vector的差集(set_difference)

差集表示在第一个vector中但不在第二个vector中的元素。同样需要排序,并使用set_difference函数。

示例代码:

// 接上面的v1, v2

vector v_diff;
v_diff.resize(max(v1.size(), v2.size())); // 安全起见分配最大长度

auto it_diff = set_difference(
    v1.begin(), v1.end(),
    v2.begin(), v2.end(),
    v_diff.begin()
);

v_diff.erase(it_diff, v_diff.end());

cout for (int n : v_diff) cout cout

输出:
v1 - v2 的差集: 1 6

3. 注意事项与技巧

使用这些函数时要注意以下几点:

  • 输入的vector必须是有序的,否则结果不正确
  • 目标vector需要预先分配足够空间,或使用back_inserter避免手动调整大小
  • 若不想修改原vector,可创建副本再排序
  • 对于自定义类型,需提供比较函数或重载

使用back_inserter简化操作:

vector result;
set_intersection(v1.begin(), v1.end(),
             v2.begin(), v2.end(),
             back_inserter(result));
// 无需resize,自动插入

基本上就这些。只要记得排序和空间处理,用STL算交集和差集很高效。