难度简单263收藏分享切换为英文关注反馈
给定两个数组,编写一个函数来计算它们的交集。
示例 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() >nums2.size()){ // 在这里我们设置一下数据,少的那个找出来
return intersect(nums2,nums1);// 交换数据
}
unordered_multiset<int> hash; //在这里我们选择multiset
for(auto x : nums1){
hash.insert(x);//插入数据
}
vector<int> result;
for(auto x : nums2){
if(hash.count(x)>0){//如果数据已经存在的话
result.push_back(x); //保存数据
auto it = hash.find(x);// 删除掉其中一个数据
hash.erase(it);
}
}
return result; //返回结果
}
};
// 加油加油... Try to make yourself more excellent...