两个数组的交集
题目: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)