两个数组的交集
题目
给定两个数组 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] 也是可通过的
思路
首先当需要在集合里面判断一个元素是否出现过,优先考虑哈希表。这里我们使用unordered_set,因为不需要排序且要去重。
我们定义两个unordered_set,一个用于装入num1的数据,另外一个用于保存交集。然后通过遍历nun2的数据在第一个unordered_set中找出现过的数据,如果没有找到nums.find(num) 返回的是迭代器nums.end(),找到就直接存储起来。
最后返回的是一个数组,所以使用vector转换一下。
C++代码
c
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> res;//用于存放交集
unordered_set<int> nums(nums1.begin(), nums1.end());//放入num1数组的值并去重
for(int num : nums2){
//如果存在交集,则放入res中
if(nums.find(num) != nums.end()) res.insert(num);
}
//结果要返回数组
return vector<int>(res.begin(), res.end());
}
};