哈希-力扣350. 两个数组的交集Ⅱ

news2024/9/21 2:33:10

题目

给你两个整数数组 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

思路

用map将数字对应出现的次数存成键值对，并通过大容器的遍历，将大容器中的数字逐个对照，双方都有那么就插入一个并且双方都减（1循环跳过2value--），如果有个没了那么就不让查找，说明不存在该交集，小存在map中，防止循环不充分。

代码实现

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
    	if(nums1.size()>nums2.size()){
			return intersect(nums2,nums1);//防止遍历不完全，将小数组放到map中
		}
		unordered_map<int,int>m;
		for(int num:nums1){
			m[num]++;//将小容器中的数字加入到m中并以value计数
		}
		vector<int>ans;//结果
		for(int num:nums2){//遍历
			if(m.count(num)){//如果存在这个键说明这时nums1中还存在该元素
				ans.emplace_back(num);存在一个那么ans加一个
				m[num]--;//减少一个（交集）
				if(!m[num])//一旦发现另一个没有了，那么移除该元素，已经不存在该交集了
				   m.erase(num);//移除
			}
		}
		return ans;//返回该容器
    }
};