给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:
输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
进阶:
如果给定的数组已经排好序呢?你将如何优化你的算法?
如果 nums1 的大小比 nums2 小很多,哪种方法更优?
如果 nums2 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
直接调用c++中算法库中的求交集的函数set_intersection,这个函数在algorithm头文件中,用这个函数求交集需要先对两个数组排序。
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
vector<int> v;
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
set_intersection(nums1.begin(),nums1.end(),nums2.begin(),nums2.end(),back_inserter(v));
return v;
}
};
61 / 61 个通过测试用例
执行用时:12 ms
对两个数组排序,然后遍历两个数组,如果当前下标下两个数组相同,则将这个元素插入到要返回的数组中。如果某个大于另外一个,元素更小的数组的小标自增。
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
vector<int> v;
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
int size1=nums1.size(),size2=nums2.size();
int i=0,j=0;
while(i<size1&&j<size2){
if(nums1[i]<nums2[j])
i++;
else if(nums1[i]>nums2[j])
j++;
else{
v.push_back(nums1[i]);
++i;++j;
}
}
return v;
}
};
61 / 61 个通过测试用例
执行用时:12 ms
使用map
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
vector<int> v;
map<int,int> m_temp;
for(auto i:nums1){
if(m_temp.find(i)==m_temp.end())
m_temp[i]=1;
else
m_temp[i]++;
}
for(int j=0;j<nums2.size();j++){
if(m_temp.find(nums2[j])!=m_temp.end() && m_temp[nums2[j]]>0){
v.push_back(nums2[j]);
m_temp[nums2[j]]--;
}
}
return v;
}
};
执行用时:16 ms