想到的解法是使用两个哈希表,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;}
};