使用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
using namespace std;
int main() {
vector
vector
vector
// 必须先排序
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
立即学习“C++免费学习笔记(深入)”;
2. 求两个vector的差集(set_difference)
差集表示在第一个vector中但不在第二个vector中的元素。同样需要排序,并使用set_difference函数。
示例代码:
// 接上面的v1, v2vector
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简化操作:
vectorset_intersection(v1.begin(), v1.end(),
v2.begin(), v2.end(),
back_inserter(result));
// 无需resize,自动插入
基本上就这些。只要记得排序和空间处理,用STL算交集和差集很高效。











