349. 两个数组的交集

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] 也是可通过的

提示：

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

Go代码

func intersection(nums1 []int, nums2 []int) []int {
    // 先将两个数组各自去重用Set存起来，然后找出两个Set中都存在的元素即可
    var res []int
    set := make(map[int]struct{})
    set2 := make(map[int]struct{})

    for _,num := range nums1 {
        set[num] = struct{}{}
    }

    for _,num := range nums2 {
        set2[num] = struct{}{}
    }


    for key := range set {
        if _,ok := set2[key];ok {
            res = append(res,key)
        }
    }

    return res
}

350. 两个数组的交集 II

350. 两个数组的交集 II

给你两个整数数组 nums1 和 nums2 ，请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数，应与元素在两个数组中都出现的次数一致（如果出现次数不一致，则考虑取较小值）。可以不考虑输出结果的顺序。

示例 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 <= nums1.length, nums2.length <= 1000
• 0 <= nums1[i], nums2[i] <= 1000

进阶：

• 如果给定的数组已经排好序呢？你将如何优化你的算法？
• 如果 nums1 的大小比 nums2 小，哪种方法更优？
• 如果 nums2 的元素存储在磁盘上，内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？

Go代码

func intersect(nums1 []int, nums2 []int) []int {
    // 由于题目已经限定了两个数组的长度和数字都在0-1000以内(共1001个数），所以可以用数组充当Map
    // 下标索引作为key，对应数字值，对应数组位置的元素为Value，即数字出现的次数
    
    res := make([]int,0)
    arr1,arr2 := make([]int,1001),make([]int,1001)
    for i := 0;i < len(nums1);i++ {
        arr1[nums1[i]]++
    }

    for i:=0;i < len(nums2);i++ {
        arr2[nums2[i]]++
    }

    // 遍历两个数组，同一位置数字都不为0则是一个交集
    for i := 0; i < len(arr1);i++ {
        if arr1[i] != 0 && arr2[i] != 0 {
            // 元素在两个数组中都出现的次数如果不一致，则考虑取较小值,添加对应次数进交集
            for j := 0;j < getMin(arr1[i],arr2[i]);j++{
                res = append(res,i)
            }
        }
    }

    return res
}

func getMin(a,b int) int {
    if a < b {
        return a
    }
    return b
}