349.两个数组的交集

Aug 6, 2024

练习次数 **

给定两个数组 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] 也是可通过的

提示：

1 <= nums1.length, nums2.length <= 1000 0 <= nums1[i], nums2[i] <= 1000

题解

func intersection(nums1 []int, nums2 []int) []int {
    result := []int{}
    resultMap := map[int]int{}
    for _, v := range nums1 {
        if _, ok := resultMap[v]; !ok {
            resultMap[v] = 1
        }
    }

    for _, v := range nums2 {
        if value, ok := resultMap[v]; ok && value != 0 {
            result = append(result, v)
            resultMap[v] = 0
        }
    }

    return result
}