使用set_intersection和set_difference前需排序两vector,分别求交集与差集,结果存入预分配或通过back_inserter动态扩展的容器中。
在C++中,求两个vector的交集和差集可以通过标准库中的算法函数实现,前提是这两个vector已经排序。主要使用头文件中的set_intersection和set_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;
se
t_intersection(v1.begin(), v1.end(),
v2.begin(), v2.end(),
back_inserter(result));
// 无需resize,自动插入
基本上就这些。只要记得排序和空间处理,用STL算交集和差集很高效。