题目描述
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
思路分析
- 计算重复的交集,我们可以采用一个数组或者map集合来记录其中一个数组出现的次数
- 遍历另一个数组,得到上一个数组出现过的数,并记录到一个新的切片中即可
代码实现
/**
* 声明map,记录较短的数组,之后遍历长数组
* 如果map中有该值对应的键且值大于零,将该值减一,如果为0,那么就清除这个键值对
*/
func intersect(nums1 []int, nums2 []int) []int {
var numsMap map[int]int
numsMap = make(map[int]int, len(nums1))
for i := 0; i < len(nums1); i++ {
item, ok := numsMap[nums1[i]]
if ok {
numsMap[nums1[i]] = item + 1
} else {
numsMap[nums1[i]] = 1
}
}
var res []int
if len(numsMap) == 0 {
return res
}
for j := 0; j < len(nums2); j++ {
item, ok := numsMap[nums2[j]]
if ok {
if item > 0 {
numsMap[nums2[j]] = item - 1
res = append(res, nums2[j])
}
}
}
return res
}