71、【哈希表】leetcode——350. 两个数组的交集 II（C++/Python版本）

辰阳星宇

已于 2024-07-29 22:03:17 修改

阅读量100

点赞数

分类专栏： # 哈希表数据结构与算法刷题文章标签： leetcode c++ 算法哈希表

于 2023-01-02 18:50:41 首次发布

本文链接：https://blog.csdn.net/qq_41094332/article/details/128523483

版权

数据结构与算法刷题同时被 2 个专栏收录

215 篇文章 16 订阅

订阅专栏

哈希表

15 篇文章 0 订阅

订阅专栏

该文介绍了如何利用C++中的unordered_map数据结构解决两个数组交集的问题。在遍历nums1时，将元素作为键并增加计数，然后遍历nums2，如果键存在于map中且值不为0，则将元素添加到结果中并减少计数。这种方法的时间复杂度和空间复杂度均为O(n)。

摘要由CSDN通过智能技术生成

题目描述

在这里插入图片描述

原题链接：350. 两个数组的交集 II

解题思路

因为交集中不去重，可存在重复元素，因此采用unordered_map，将元素作为下标Key，每有一个元素，则对应的Value加一。寻找交集时，遍历存储过的Hash表，Value不为0时，就将其加入到res中，然后将该Key对应的Value减一。

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        unordered_map<int, int> record;
        for(int num : nums1) {
            record[num]++;            
        }
        for(int i = 0; i < nums2.size(); i++) {
            if(record[nums2[i]] != 0) {
                res.push_back(nums2[i]);
                record[nums2[i]]--;
            }
        }
        return res;
    }
};

Python

class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        nums1_dict = Counter(nums1)
        res = []
        for num in nums2:
            if num in nums1_dict.keys():
                res.append(num)
                nums1_dict[num] -= 1
                if nums1_dict[num] == 0:
                    del nums1_dict[num]
        
        return res