链接: https://leetcode-cn.com/problems/intersection-of-two-arrays-ii
给定两个数组,编写一个函数来计算它们的交集。
说明
输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
示例 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 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
思路一: 排序
先对两个数组分别进行排序, 设两个指针p1和p2, 初始时分别指向第一个元素, 然后比较两个指针指向元素的值, 若*p1小于*p2, 则++p1, *p2小于*p1, 则++p2, 若两个相等, 则保存此值, 然后p1,p2分别自增。若其中一个数组遍历完成, 则程序结束。
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
vector<int> result;
std::sort(nums1.begin(),nums1.end());
std::sort(nums2.begin(),nums2.end());
vector<int>::iterator it1 = nums1.begin();
vector<int>::iterator it2 = nums2.begin();
while(it1<nums1.end() && it2<nums2.end()) {
if((*it1) < (*it2)) {
++it1;
} else if ((*it1) > (*it2)) {
++it2;
} else {
result.push_back(*it1);
++it1, ++it2;
}
}
return result;
}
思路二: Hash
我们先在 HashMap 记录一个数组中的存在的数字和对应出现的次数。然后,我们遍历第二个数组,检查数字在 HashMap 中是否存在,如果存在且计数为正,则将该数字添加到答案并减少 HashMap 中的计数。
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
if (nums1.size() > nums2.size()) {
return intersect(nums2, nums1);
}
unordered_map<int, int> m;
for (auto n : nums1) {
++m[n];
}
int k = 0;
for (auto n : nums2) {
auto it = m.find(n);
if (it != end(m) && --it->second >= 0) {
nums1[k++] = n;
}
}
return vector(begin(nums1), begin(nums1) + k);
}
思路三:
这类似于思路1。我们不使用两个指针进行迭代,而是使用内置函数来查找公共元素。在 C++ 中,我们可以使用 set_intersection 来排序数组(或 multisets)。
在 Java 中的 retainAll 方法并不关心一个元素在另一个集合中出现的次数。
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
sort(begin(nums1), end(nums1));
sort(begin(nums2), end(nums2));
nums1.erase(set_intersection(begin(nums1), end(nums1),
begin(nums2), end(nums2), begin(nums1)), end(nums1));
return nums1;
}