一、题目内容
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2] 输出:[2,2]
示例 2:输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出:[4,9]
说明:
- 输出结果中每个元素出现的次数,应与元素在两个数组中出现次数的最小值一致。
- 我们可以不考虑输出结果的顺序。
进阶:
- 如果给定的数组已经排好序呢?你将如何优化你的算法?
- 如果 nums1 的大小比 nums2 小很多,哪种方法更优?
- 如果 nums2 的元素存储在磁盘上,内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
二、解题思路
首先拿到这道题,我们基本马上可以想到,此题可以看成是一道传统的映射题(map映射),为
什么可以这样看呢,因为我们需找出两个数组的交集元素,同时应与两个数组中出现的次数一
致。这样就导致了我们需要知道每个值出现的次数,所以映射关系就成了<元素,出现次数>。剩下
的就是顺利成章的解题。
三、代码实现
func intersect(nums1 []int, nums2 []int) []int {
m0 := map[int]int{}
for _, v := range nums1 {
//遍历nums1,初始化map
m0[v] += 1
}
k := 0
for _, v := range nums2 {
//如果元素相同,将其存入nums2中,并将出现次数减1
if m0[v] > 0 {
m0[v] -=1
nums2[k] = v
k++
}
}
return nums2[0:k]
}