LeetCode 2956.找到两个数组中的公共元素：哈希表

【LetMeFly】2956.找到两个数组中的公共元素：哈希表

力扣题目链接：https://leetcode.cn/problems/find-common-elements-between-two-arrays/

给你两个下标从 0 开始的整数数组 nums1 和 nums2 ，它们分别含有 n 和 m 个元素。

请你计算以下两个数值：

• 统计 0 <= i < n 中的下标 i ，满足 nums1[i] 在 nums2 中 至少 出现了一次。
• 统计 0 <= i < m 中的下标 i ，满足 nums2[i] 在 nums1 中 至少 出现了一次。

请你返回一个长度为 2 的整数数组 answer ，按顺序 分别为以上两个数值。

示例 1：

输入：nums1 = [4,3,2,3,1], nums2 = [2,2,5,2,3,6]
输出：[3,4]
解释：分别计算两个数值：
- nums1 中下标为 1 ，2 和 3 的元素在 nums2 中至少出现了一次，所以第一个值为 3 。
- nums2 中下标为 0 ，1 ，3 和 4 的元素在 nums1 中至少出现了一次，所以第二个值为 4 。

示例 2：

输入：nums1 = [3,4,2,3], nums2 = [1,5]
输出：[0,0]
解释：两个数组中没有公共元素，所以两个值都为 0 。

提示：

• n == nums1.length
• m == nums2.length
• 1 <= n, m <= 100
• 1 <= nums1[i], nums2[i] <= 100

解题方法：哈希表

首先将两个数组中的元素放入哈希表中，这样就能在$O(1)$时间内判断一个元素有没有在一个数组中。

接着分别遍历两个数组中，统计有多少个元素在另一个数组中出现过。

• 时间复杂度$O(len(nums1)+len(nums2))$
• 空间复杂度$O(len(nums1)+len(nums2))$

AC代码

C++

class Solution {
public:
    vector<int> findIntersectionValues(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> s1(nums1.begin(), nums1.end()), s2(nums2.begin(), nums2.end());
        vector<int> ans(2);
        for (int t : nums1) {
            ans[0] += s2.count(t);
        }
        for (int t : nums2) {
            ans[1] += s1.count(t);
        }
        return ans;
    }
};

Go

package main

func findIntersectionValues(nums1 []int, nums2 []int) []int {
    s1, s2 := map[int]int{}, map[int]int{}
    for _, t := range nums1 {
        s1[t] = 1
    }
    for _, t := range nums2 {
        s2[t] = 1
    }
    ans := [2]int{}
    for _, t := range nums1 {
        ans[0] += s2[t]
    }
    for _, t := range nums2 {
        ans[1] += s1[t]
    }
    return ans[:]
}

Java

import java.util.HashSet;

class Solution {
    public int[] findIntersectionValues(int[] nums1, int[] nums2) {
        HashSet<Integer> s1 = new HashSet<>(), s2 = new HashSet<>();
        for (int t : nums1) {
            s1.add(t);
        }
        for (int t : nums2) {
            s2.add(t);
        }
        int[] ans = new int[2];
        for (int t : nums1) {
            ans[0] += s2.contains(t) ? 1 : 0;
        }
        for (int t : nums2) {
            ans[1] += s1.contains(t) ? 1 : 0;
        }
        return ans;
    }
}

Python

from typing import List

class Solution:
    def findIntersectionValues(self, nums1: List[int], nums2: List[int]) -> List[int]:
        return [sum(t in nums2 for t in nums1), sum(t in nums1 for t in nums2)]

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

Tisfy：https://letmefly.blog.csdn.net/article/details/140479914