题目:Intersection of Two Arrays
原题链接
Given two arrays, write a function to compute their intersection.
Example:
Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].
Note:
Each element in the result must be unique.
The result can be in any order.
给出两个数组,找出他们这个两个数组中共有的元素,要求结果中存放的元素唯一,顺序可以任意。
方法一:分别将两个数组排序,然后从头到尾扫描扫描两个数组,要是有相同元素就插入,由于数组已经有序,所以当向结果中插入元素的时候只需要判断和结果中的最后一个元素是不是相等就行了。代码如下:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
sort(nums1.begin(), nums1.end());
sort(nums2.begin(), nums2.end());
vector<int> ans;
int i = 0, j = 0, k = 0;
int len1 = nums1.size(), len2 = nums2.size();
while((i < len1) && (j < len2)){
if(nums1[i] == nums2[j] && (!k || nums1[i] != ans[k - 1])){
ans.push_back(nums1[i]);
++i, ++j, ++k;
}else if(nums1[i] < nums2[j]){
++i;
}else if(nums1[i] > nums2[j]){
++j;
}else{
++i, ++j;
}
}
return ans;
}
};
方法二:用无序集合(unordered_set),这样可以直接过滤掉重复的元素,然后挨个查找数组2中的元素就可以了,代码如下:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> se(nums1.begin(),nums1.end());
vector<int> ans;
for(int num : nums2){
if(se.count(num)){
ans.push_back(num);
se.erase(num);
}
}
return ans;
}
};