LeetCode 349. 两个数组的交集

题目描述

349. 两个数组的交集

给定两个数组 nums1 和 nums2 ，返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。

示例 1：

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

示例 2：

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]
解释：[4,9] 也是可通过的

代码

# go

func intersection(nums1 []int, nums2 []int) []int {
    numMap := make(map[int]struct{})
    // 将nums1不同的数放入map
    for _, val := range nums1 {
        if _, ok := numMap[val]; !ok {
            numMap[val] = struct{}{}
        }
    }

    ret := []int{}
    // 遍历nums2的数，如果在map中找到则加入数据，并从map中删除
    for _, val := range nums2 {
        if _, ok := numMap[val]; ok {
            // 不要用ret.append直接用append
            // 否则会报type []int has no field or method append
            ret = append(ret, val)
            delete(numMap, val)
        }
    }

    return ret
}