想到的解法是使用两个哈希表,s1用来统计nums1中出现过的数字,然后遍历nums2数组,当能够在s1中查找到nums2的元素时,将这个元素添加到s2中,最后遍历s2,将其中的元素添加到返回数组中。
但最开始写时,我在第二个for循环内,判断nums2[j]是否在s1中,使用的不是find函数,而是 让nums2[j]插入s1,根据insert的返回值 pair.second 来判断是否存在,此时就出现了一个问题。当s2中存在两个相同的元素时,不论s1是否存在,这个元素都将会被添加两次,这样的写法是行不通的。
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> s1;
unordered_set<int> s2;
vector<int> v;
for(int i = 0; i < nums1.size(); i++){
s1.insert(nums1[i]);
}
for(int j = 0; j < nums2.size(); j++){
if(s1.find(nums2[j]) != s1.end()){
s2.insert(nums2[j]);
}
}
for(auto num : s2){
v.push_back(num);
}
return v;
}
};