两个数组的交集 II——Go

题目描述

给定两个数组，编写一个函数来计算它们的交集。

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]

示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]

思路分析

1. 计算重复的交集，我们可以采用一个数组或者map集合来记录其中一个数组出现的次数
2. 遍历另一个数组，得到上一个数组出现过的数，并记录到一个新的切片中即可

代码实现

/**
* 声明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
}