C++ STL 提供求交集的函数 set_intersection( ) 、求集合差的函数 set_difference( ) 和合并两个集合的函数 set_union( ) 。以 set_intersection( ) 为例,分析两端程序并简单记录其用法:
nums1:1,2,2,1
nums2:2,2
#include <algorithm> // set_intersection
#include <iterator> // insert_iterator
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
vector<int> res;
set<int> cod1(nums1.begin(),nums1.end());
set<int> cod2(nums2.begin(),nums2.end());
std::set_intersection(cod1.begin(),cod1.end(),cod2.begin(),cod2.end(),insert_iterator<vector<int>>(res,res.begin()));
return res; // res:2
}
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
vector<int> res;
std::sort(nums1.begin(),nums1.end());
std::sort(nums2.begin(),nums2.end());
std::set_intersection(nums1.begin(),nums1.end(),nums2.begin(),nums2.end(),insert_iterator<vector<int>>(res,res.begin()));
return res; // res:2,2
}
首先传递的容器必须是排序的,set 容器中元素默认是排序的,而 vector 需要调用 sort 函数进行排序。其次 set_intersection( )中最后存放交集的容器的容量必须要足够大到能放下所有的元素,即函数只执行复制,不是插入!但是模板 insert_iterator 可以将复制转换为插入,可以解决该问题。
set_intersection( ) 不是 set 的方法,而是一个通用函数,而 set 函数必须要满足这些算法!使用 set 时,可以自动忽略重复的元素,而使用 vector 时可以保留重复的元素,即保留了‘个数’这一信息。
该函数的时间复杂度为线性复杂度,其实现为:
template <class InputIterator1, class InputIterator2, class OutputIterator>
OutputIterator set_intersection (InputIterator1 first1, InputIterator1 last1,
InputIterator2 first2, InputIterator2 last2,
OutputIterator result)
{
while (first1!=last1 && first2!=last2)
{
if (*first1<*first2) ++first1;
else if (*first2<*first1) ++first2;
else {
*result = *first1;
++result; ++first1; ++first2;
}
}
return result;
}