LeetCode【学习计划】:【数据结构】
350. 两个数组的交集 II
LeetCode: 350. 两个数组的交集 II
简 单 \color{#00AF9B}{简单} 简单
给你两个整数数组
nums1
和nums2
,请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数,应与元素在两个数组中都出现的次数一致(如果出现次数不一致,则考虑取较小值)。可以不考虑输出结果的顺序。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2,2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[4,9]
提示:
1 <= nums1.length, nums2.length <= 1000
0 <= nums1[i], nums2[i] <= 1000
进阶:
- 如果给定的数组已经排好序呢?你将如何优化你的算法?
- 如果
nums1
的大小比nums2
小,哪种方法更优? - 如果
nums2
的元素存储在磁盘上,内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
方法1:哈希表
我们可以先遍历较短的数组并把其内的所有元素存入哈希表中。由于数组内存在重复的元素,为了使多个重复元素按照在2个数组中出现的次数匹配,而不至于全部匹配,所以哈希表记录的是短数组中每个字符的计数。然后遍历一次长数组。如果长数组的某个元素已经存在于哈希表了,就把它添加到答案数组。因为成功匹配到了,所以还要将哈希表中对应字符的计数减1。
设 A A A为短数组中元素的集合(包含重复元素),则哈希表在最初的状态即为 A A A。到了最后,由于短数组 A A A和长数组 B B B的交集已经摘掉了,因此最后哈希表的状态为 A − ( A ∩ B ) A-(A \cap B) A−(A∩B)。
之所以使用较短的数组存入哈希表是为了尽可能的减少实际的空间消耗。
#include <vector>
#include <unordered_map>
using namespace std;
class Solution
{
public:
vector<int> intersect(vector<int> &nums1, vector<int> &nums2)
{
bool bigger = nums1.size() > nums2.size();
unordered_map<int