349. 两个数组的交集
力扣链接
349. 两个数组的交集
给定两个数组 nums1 和 nums2 ,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
解释:[4,9] 也是可通过的
思路
set有三种
set:红黑树、有序,无重复
multiset:红黑树、有序,有重复
unordered_set:哈希表,无序,无重复
选择数据结构
从条件中看:不重复,可以无序
选择:unordered_set
方法
- 用nums1构建哈希表num_set
- 判断nums2中的元素在哈希表num_set中是否存在
- 返回数组(不是哈希表)
代码
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> result;
unordered_set<int> num_set(nums1.begin(),nums1.end());
for(int num:nums2)
{
if(num_set.find(num)!=num_set.end())//num_set.end()指向是空,说明搜索的num在表中
result.insert(num);
}
return vector<int>(result.begin(),result.end());
}
};