350-两个数组的交集
访问量:920

一、题目

题目地址:https://leetcode.com/problems/intersection-of-two-arrays-ii/submissions/

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

示例 1:

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

示例 2:

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

说明:

  • 输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。

  • 我们可以不考虑输出结果的顺序。

二、解法

func intersect(nums1 []int, nums2 []int) []int {
    var rs []int
    if len(nums1) > len(nums2) {
        num3 := nums2
        nums2 = nums1
        nums1 = num3
    }

    var numsTimes = make(map[int]int)
    for _, val :=range nums1 {
        numsTimes[val] ++
    }

    for _,val :=range nums2 {
        if  curTimes, ok :=numsTimes[val]; ok{
            if curTimes == 0 {
                //删除该元素
                delete(numsTimes, val)
                continue
            }
            numsTimes[val] --
            rs = append(rs, val)
        } else { //不存在的元素
            continue
        }
    }

    return  rs
}