给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
说明:
输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。
来源:力扣(LeetCode)
链接:leetcode-349-两个数组的交集
思路:
1.先利用set集合将数组元素去重
2.利用迭代器,遍历其中一个set,如果元素在另一个set中,则保存下来。
代码:
class Solution {
public:vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {set<int>set1(nums1.begin(),nums1.end());set<int>set2(nums2.begin(),nums2.end());vector<int>ret;set<int>::iterator iter;//建立一个set<int>集合类型的迭代器for(iter=set1.begin();iter!=set1.end();iter++){if(find(set2.begin(),set2.end(),*iter)!=set2.end()){ret.push_back(*iter);}}return ret;}
};
可以发现,貌似用数组做哈希表可以解决这道题目,把nums1的元素,映射到哈希数组的下表上,然后在遍历nums2的时候,判断是否出现过就可以了。
但是要注意,「使用数据来做哈希的题目,都限制了数值的大小,例如哈希表:可以拿数组当哈希表来用,但哈希值不要太大题目中只有小写字母,或者数值大小在[0- 10000] 之内等等。」
而这道题目没有限制数值的大小,就无法使用数组来做哈希表了。
「而且如果哈希值比较少、特别分散、跨度非常大,使用数组就造成空间的极大浪费。」
此时就要使用另一种结构体了,set ,关于set,C++ 给提供了如下三种可用的数据结构:
std::set
std::multiset
std::unordered_set
std::set和std::multiset底层实现都是红黑树,std::unordered_set的底层实现是哈希表, 使用unordered_set 读写效率是最高的,并不需要对数据进行排序,而且还不要让数据重复,所以选择unordered_set。
class Solution {
public:vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {unordered_set<int> set1(nums1.begin(),nums1.end());unordered_set<int> result_set;for(auto num: nums2){if(set1.find(num) != set1.end())result_set.insert(num);}return vector<int>(result_set.begin(), result_set.end());}
};