2956. 找到两个数组中的公共元素
题目描述:
给你两个下标从 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
解题分析及思路:
方法:哈希表
思路:
将数组中的元素作为键,出现次数作为值,分别创建两个哈希表,分别统计两个数组中每个元素的出现次数。
然后遍历其中一个哈希表,如果另一个哈希表中存在相同的键,则将两个哈希表中对应键的值相加。
最后返回两个数值组成的数组。
func findIntersectionValues(nums1 []int, nums2 []int) []int {
var nums1Map = make(map[int]int)
var nums2Map = make(map[int]int)
for _, num := range nums1 {
nums1Map[num]++
}
for _, num := range nums2 {
nums2Map[num]++
}
var result = make([]int, 2)
for num, count := range nums1Map {
if _, ok := nums2Map[num]; ok {
result[0] += count
result[1] += nums2Map[num]
}
}
return result
}
复杂度:
- 时间复杂度:O(m + n),其中 m 和 n 分别为 nums1 和 nums2 的长度。
- 空间复杂度:O(m + n)
执行结果:
- 执行耗时:19 ms,击败了50.00% 的Go用户
- 内存消耗:6.6 MB,击败了34.38% 的Go用户