题目描述:
给定两个数组,写一个函数来计算它们的交集。
例子:
给定 num1= [1, 2, 2, 1], nums2 = [2, 2], 返回 [2].
提示:
每个在结果中的元素必定是唯一的。
我们可以不考虑输出结果的顺序。
题目分析:
要找到两个数组的交集,先对两个数组进行排序,之后从每个数组的开始进行遍历,如果是公共元素,那么放入set容器当中。
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
vector<int>ans;
set<int>st;
if(nums1.size()==0||nums2.size()==0){
return ans;
}
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
int i=0,j=0,len=0;
while(i<nums1.size()&&j<nums2.size()){
if(nums1[i]==nums2[j]){
st.insert(nums1[i]);
i++;j++;
}
else if(nums1[i]<nums2[j]){
i++;
}
else{
j++;
}
}
for(set<int>::iterator it=st.begin();it!=st.end();it++){
ans.push_back(*it);
}
return ans;
}
};