题目类型
已知两个或者多个集合,求他们的交集。
也就是集合之间公共的元素。
解题方法
并查集:
利用并查集,对集合中出现的各种元素进行数量统计。然后再查询验证,选出每个集合都出现过的元素。
这里会有一个是使用map还是undered_map的选择,可以参考一下这篇文章c++中map与unordered_map的区别
排序+双指针
先将集合中的元素(一般是数字)进行排序,然后通过指针对集合进行遍历。如果指针指向的元素相同,即为交集中的元素,如果不相同,则将指针后移。
两种方法的比较
如果内存有限,当集合比较大时,无法将其元素全部读取到内存中,这时就应该采用并查集,因为用并查集进行查询时,我们不用将集合中所有元素全度读取出来,只需要在验证的时候再读取就行了。
但是排序的话就不行了,需要将所有元素全部读取到内存中才能进行排序。
不过两种方法都只适合只有两三个集合的情况,如果集合过多,使用这两种方法会比较麻烦。
例题
上面的方法说明有些抽象,用一个例题具体说明吧!
力扣350.两个数组的交集 II
并查集:
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
//保证nums1长度比nums2长度要小
if(nums1.size() > nums2.size())
return intersect(nums2,nums1);
//创建并查集
unordered_map<int,int> m;
//创建向量存储答案
vector<int> ans;
//先用一个循环遍历nums1,统计在nums1中出现过的数字
for(int num : nums1){
m[num]++;
}
//再用一个循环,遍历nums2
for(int num : nums2){
//判断nums中的数字在nums1中是否出现
if(m[num])
{
ans.push_back(num);
m[num]--;//保证出现次数相同
}
if(m[num]==0)
m.erase(num);
}
return ans;
}
};
排序+双指针:
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
int len1 = nums1.size();
int len2 = nums2.size();
//创建向量存放答案
vector<int> ans;
if(len1 == 0 || len2 == 0) return ans;
//利用sort函数对nums1和nums2进行排序
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
//创建两个指针
int index1 = 0, index2 = 0;
//遍历两个集合
while(index1 < len1 && index2 < len2){
//当nums1[index1]比nums2[index2]大时,指针一向后移动一位
if(nums1[index1] < nums2[index2]){
index1++;
}
//当nums1[index1]比nums2[index2]小时,指针二向后移动一位
else if(nums1[index1] > nums2[index2]){
index2++;
}
//当nums1[index1]比nums2[index2]相等时,将数字存入ans中,然后将两个指针都向后移动一位
else{
ans.push_back(nums1[index1]);
index1++;
index2++;
}
}
return ans;
}
};
类似的题目还有力扣349.349. 两个数组的交集
这题与上一题的区别就是不考虑重复的元素。