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.
Subscribe to see which companies asked this question.
Show Similar Problems
题目的意思:找出两个集合的交集,并且交集中没有重复的内容。题目要求输出的结果可以是任意顺序。
我的思路跟上一题一样,只不过在得到交集之后,再将重复的元素删除;但是答案的结果不是按照从小到大的顺序排列的,所以跟答案对不上。但是为了保存,我把我的按顺序输出的代码先放在这里。
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
vector<int> res;
vector<int>::iterator it;
if(nums1.size()>=nums2.size())
{
for(int i=0;i<nums2.size();i++)
{
int left=0;
int right=nums1.size()-1;
while(left<=right)
{
int mid=(right-left)/2+left;
if(nums2[i]==nums1[mid])
{
res.push_back(nums2[i]);
it=nums1.begin()+mid;
nums1.erase(it);
break;
}
else if(nums2[i]>nums1[mid])
left=mid+1;
else
right=mid-1;
}
}}
else if(nums1.size()<nums2.size())
{
for(int i=0;i<nums1.size();i++)
{
int left=0;
int right=nums2.size()-1;
while(left<=right)
{
int mid=(right-left)/2+left;
if(nums1[i]==nums2[mid])
{
res.push_back(nums1[i]);
it=nums2.begin()+mid;
nums2.erase(it);
break;
}
else if(nums1[i]>nums2[mid])
left=mid+1;
else
right=mid-1;
}
}
}
for(int j=1;j<res.size();j++)
{
if(res[j]==res[j-1])
{it=res.begin()+j;
res.erase(it);}
}
return res;
}
};
经过修改之后,符合最终的答案
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
vector<int> res;
vector<int>::iterator it;
for(int i=0;i<nums2.size();i++)
{
for(int j=0;j<nums1.size();j++)
{
if(nums2[i]==nums1[j])
{
res.push_back(nums2[i]);
it=nums1.begin()+j;
nums1.erase(it);
break;
}
}
}
for(int j=0;j<res.size();j++)
{
for(int i=j+1;i<res.size();i++)
if(res[j]==res[i])
{it=res.begin()+i;
res.erase(it);}
}
return res;
}
};
注意的是:因为不是按顺序进行排列的,所以最后一步,去掉容器中重复的数的时候,要遍历完所有的数字,而不仅仅是相邻两个数字。
看到讨论中用c++11新特性中的无序关联容器unorderd_set来做,很简洁
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> m(nums1.begin(), nums1.end());
vector<int> res;
for (auto a : nums2)//a为num2中任意的值
if (m.count(a)) {//计算m中a出现的次数,如果a出现的次数大于1就为真
res.push_back(a);
m.erase(a);//将m中值为a的值都删掉
}
return res;
}
};