搬：关于 std::set_intersection( ) 函数用法

【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

1. vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {

2. vector<int> res;

3. set<int> cod1(nums1.begin(),nums1.end());

4. set<int> cod2(nums2.begin(),nums2.end());

5. std::set_intersection(cod1.begin(),cod1.end(),cod2.begin(),cod2.end(),insert_iterator<vector<int>>(res,res.begin()));

6. return res; // res：2

7. }

1. vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {

2. vector<int> res;

3. std::sort(nums1.begin(),nums1.end());

4. std::sort(nums2.begin(),nums2.end());

5. std::set_intersection(nums1.begin(),nums1.end(),nums2.begin(),nums2.end(),insert_iterator<vector<int>>(res,res.begin()));

6. return res; // res：2，2

7. }

首先传递的容器必须是排序的，set 容器中元素默认是排序的，而 vector 需要调用 sort 函数进行排序。其次 set_intersection( )中最后存放交集的容器的容量必须要足够大到能放下所有的元素，即函数只执行复制，不是插入！但是模板 insert_iterator 可以将复制转换为插入，可以解决该问题。

 set_intersection( ) 不是 set 的方法，而是一个通用函数，而 set 函数必须要满足这些算法！使用 set 时，可以自动忽略重复的元素，而使用 vector 时可以保留重复的元素，即保留了‘个数’这一信息。

该函数的时间复杂度为线性复杂度，其实现为：

1. template <class InputIterator1, class InputIterator2, class OutputIterator>

2. OutputIterator set_intersection (InputIterator1 first1, InputIterator1 last1,

3. InputIterator2 first2, InputIterator2 last2,

4. OutputIterator result)

5. {

6. while (first1!=last1 && first2!=last2)

7. {

8. if (*first1<*first2) ++first1;

9. else if (*first2<*first1) ++first2;

10. else {

11. *result = *first1;

12. ++result; ++first1; ++first2;

13. }

14. }

15. return result;

16. }