【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(1) O(1)时间内判断一个元素有没有在一个数组中。
接着分别遍历两个数组中,统计有多少个元素在另一个数组中出现过。
- 时间复杂度 O ( l e n ( n u m s 1 ) + l e n ( n u m s 2 ) ) O(len(nums1) + len(nums2)) O(len(nums1)+len(nums2))
- 空间复杂度 O ( l e n ( n u m s 1 ) + l e n ( n u m s 2 ) ) 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