力扣:350.两个数组的交集 II
题目:
给你两个整数数组 nums1 和 nums2 ,请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数,应与元素在两个数组中都出现的次数一致(如果出现次数不一致,则考虑取较小值)。可以不考虑输出结果的顺序。
思路:
查找值的同时,若满足则还要比较此元素在两容器中出现的次数,多了则不填入目标容器中。
次数比较的实现:
次数通过先记录元素在第一个数组中出现的次数,然后若在第二个数组中出现则次数–,次数值为0即大小比较的边界
首先记录第一个数组的每个元素及出现的次数,然后依次判断第二个数组中的元素是否存在于第一个数组中,若存在首先判断其出现的个数是否小于等于0,若小于等于0则表示此时判断的第二个数组中的这个数不能再添加进结果容器中了。
代码:
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
if(nums1.size() > nums2.size()){
return intersect(nums2,nums1);
}
unordered_map<int, int>m;
for(int num:nums1){
++m[num];
}
vector<int> v;
for(int num:nums2){
if(m.count(num)){
v.push_back(num);
--m[num];
if(m[num] == 0) m.erase(num);
}
}return v;
}
};