给定两个数组,编写一个函数来计算它们的交集。
示例 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 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
if(nums1.size()==NULL)return nums1;
if(nums2.size()==NULL)return nums2;
//int a=nums1.size()>nums2.size()?nums2.size():nums1.size();
unordered_map<int,int>hashmap;
vector<int>result;
for(int i=0;i<nums1.size();++i){
if(hashmap.count(nums1[i]))
hashmap[nums1[i]]++;
else
hashmap.insert(make_pair(nums1[i],1));
}
for (int i=0;i<nums2.size();i++){
if(hashmap.count(nums2[i])){
result.push_back(nums2[i]);
hashmap[nums2[i]]--;
if( hashmap[nums2[i]]==0)
hashmap.erase(nums2[i]);//匹配到就把value减1,如果value减到0,就把这个key删掉。
}
}
return result;
}
};