leetcode 探索 哈希表 两个数组的交集

题目

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

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

示例 2：

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

说明：
输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。

分析

通过遍历一遍数组1，然后加入到map中。这样再遍历数组2，遇到数组1中有的，就加入结果集，同时要注意到题目要求结果集是唯一的，那么我们在遇到过之后，就要把map中该元素删除了。

解法

func intersection(nums1 []int, nums2 []int) []int {
    set1 := make(map[int]struct{}, len(nums1))
    
    for _, v := range nums1 {
        set1[v] = struct{}{}
    }
    
    var result []int
    for _, v := range nums2 {
        if _, ok := set1[v]; ok {
            result = append(result, v)
            delete(set1, v)
        }
    }
    return result
}