给定两个数组,写一个函数来计算它们的交集。
例子:
给定 num1= [1, 2, 2, 1]
, nums2 = [2, 2]
, 返回 [2]
.
提示:
- 每个在结果中的元素必定是唯一的。
- 我们可以不考虑输出结果的顺序。
题目分析:首先,将num1的元素存进s1,完成了一步去重;同样,将num2的元素存进s2,也完成了一步去重;将所有的元素存进s. 所以s=s1s2,要求的是s1
s2. 我们知道s1
s2=s2-(s-s1),即可求解。
代码展示:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
set<int> s1;
set<int> s2;
set<int> s;
for(auto num:nums1){
s1.insert(num);
s.insert(num);
}
for(auto num:nums2){
s2.insert(num);
s.insert(num);
}
for(auto temp:s1){
s.erase(temp);
}
for(auto temp:s){
s2.erase(temp);
}
vector<int> ans;
for(auto temp:s2)
ans.push_back(temp);
return ans;
}
};