当前位置：首页 > article >正文

LeetCode350：两个数组的交集Ⅱ

article 2024/11/5 16:28:03

题目描述
给你两个整数数组 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]

解题思想
使用hashMap，将第一个数组的元素插入hashMap。
然后遍历第二个数组，看其num是否在hashMap中，如果在就将count–，并插入数组中。

代码

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int,int> hash;
        vector<int> ans;
        
        for(int num:nums1)
            ++hash[num];
            
        for(int num:nums2)
            if(hash.count(num)){
                --hash[num];
                ans.push_back(num);

                if(hash[num] == 0)
                    hash.erase(num);
            }   
        return ans;
    }
};