给定两个数组,写一个方法来计算它们的交集。
例如:
给定 nums1 = [1, 2, 2, 1]
, nums2 = [2, 2]
, 返回 [2, 2]
.
注意:
- 输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
- 我们可以不考虑输出结果的顺序。
跟进:
- 如果给定的数组已经排好序呢?你将如何优化你的算法?
- 如果 nums1 的大小比 nums2 小很多,哪种方法更优?
- 如果nums2的元素存储在磁盘上,内存是有限的,你不能一次加载所有的元素到内存中,你该怎么办?
思路:
这道题要求如果有重复的数字也把重复的数字输出,所以我们用一个hashmap,key保存数组的值,value对应值出现的次数,先把nums1的所有数字放入hashmap中,再遍历nums2,如果nums2中的数字在hashmap中有保存且次数不为0,则加到res中,最后返回res。
代码如下:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
vector<int> res;
unordered_map<int, int> n1, n2;
for (int i = 0; i < nums1.size(); i++) {
n1[nums1[i]]++;
}
for (int i = 0; i < nums2.size(); i++) {
if (n1[nums2[i]]) {
n1[nums2[i]]--;
res.push_back(nums2[i]);
}
}
return res;
}
给定两个数组,写一个方法来计算它们的交集。
例如:
给定 nums1 = [1, 2, 2, 1]
, nums2 = [2, 2]
, 返回 [2, 2]
.
注意:
- 输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
- 我们可以不考虑输出结果的顺序。
跟进:
- 如果给定的数组已经排好序呢?你将如何优化你的算法?
- 如果 nums1 的大小比 nums2 小很多,哪种方法更优?
- 如果nums2的元素存储在磁盘上,内存是有限的,你不能一次加载所有的元素到内存中,你该怎么办?