给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
思路:
使用哈希集,利用其去重复的特性。首先将数组1排序后放进哈希集,然后遍历数组2.发现数组2中的数字在数组1中出现过,那么插入到结果中。当然存放结果也是用到哈希集,这样可以去重。然后返回数组的形式即可。
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> set(nums1.begin(),nums1.end()),res;
for(auto c:nums2){
if(set.count(c))
res.insert(c);
}
return vector<int>(res.begin(),res.end());
}
};