【C++】关于 std::set_intersection( ) 函数用法
2018年04月22日 22:00:47 Sim0Hayha 阅读数:881更多
【C++】关于 std::set_intersection( ) 函数用法
个人分类: 语言基础
C++ STL 提供求交集的函数 set_intersection( ) 、求集合差的函数 set_difference( ) 和合并两个集合的函数 set_union( ) 。以 set_intersection( ) 为例,分析两端程序并简单记录其用法:
nums1:1,2,2,1
nums2:2,2
-
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;
-
}