每日一题 --- 两个数组的交集[力扣][Go]

两个数组的交集

题目：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

方法一：

Set集合特性：

• 集合中的元素是唯一的

Go语言没有提供Set集合实现，不过可以使用map来模拟，利用的是map的key值的唯一性。

func intersection(nums1 []int, nums2 []int) []int {
	set := make(map[int]struct{}, 0)
	res := make([]int, 0)
	for _, i := range nums1 {
		if _, ok := set[i]; !ok {
			set[i] = struct{}{}
		}
	}
	for _, i := range nums2 {
		if _, ok := set[i]; ok {
			res = append(res, i)
			delete(set, i)
		}
	}
	return res
}

时间复杂度O(n + m)，空间复杂度O(n)