每日一题,防止痴呆 = =
一、题目大意
给定两个数组,编写一个函数来计算它们的交集。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/intersection-of-two-arrays-ii
二、题目思路以及AC代码
思路:哈希表
这题如果是先不考虑进阶方法的话,就只能是哈希表了。首先我们遍历一遍nums2,将nums2出现过的元素类似桶排的方法放到哈希表中,然后我们再遍历nums1中的元素,如果此时元素在哈希表中已经出现,那么将该元素加入到结果的vector中,这样的时间复杂度是 O(M + N),M和N分别是两个数组的长度。
进阶问题:
- 如果给定的数组排好序,那么我们就可以考虑类似归并的思路,利用双指针解决,即首先我们设定两个指针分别指向两个数组的第一个元素,然后比较两个指针指向的元素,如果元素相等,则将该元素加入结果数组,同时两个指针向后移动一个,如果元素不等,则将较小的那个元素指针向后移动一个,直到某一个指针超出最大范围停止。
- 如果nums1比nums2小很多,那么后一种方法,也就是双指针的方法更优,当然是要建立在已经排好序的基础上,不然排序就要O(NlogN + MlogM)的复杂度了。因为如果nums1比nums2小很多的话,我们大多数情况下都不会完全遍历nums2,所以时间复杂度会下降一些,而第一个方法的话,无论如何,其都需要全部遍历两个数组。
- 如果不能一次性读入,那就先对于每个子数组求交集,最后求并集呗。
AC代码
这里只有最开始AC的代码,那个双指针的也比较好实现,就和归并排序其实是一样的,这里就不特殊写了。
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
vector<int> res;
unordered_map<int, int> vis2;
for (auto num: nums2) {
vis2[num]++;
}
for (int i=0;i<nums1.size();i++) {
if (vis2[nums1[i]]) {
res.push_back(nums1[i]);
vis2[nums1[i]]--;
}
}
return res;
}
};
如果有问题,欢迎大家指正!!!