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