LeetCode-349.两个数组的交集

哈希表#

给定两个数组 nums1 和 nums2 ，返回 它们的交集 。输出结果中的每个元素一定是唯一的。我们可以 不考虑输出结果的顺序 。

示例 1：

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

示例 2：

1
输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
2
输出：[9,4]
3
解释：[4,9] 也是可通过的

1.自己的解法（在242.有效的字母异位词的基础上修改）

1
func intersection(nums1 []int, nums2 []int) []int {
2
    record1 := make(map[int]int)
3
    for _, i := range nums1{
4
        record1[i]++
5
    }
6

7
    record2 := make(map[int]int)
8
    for _, i := range nums2{
9
        record2[i]++
10
    }
11

12
    output := []int{}
13
    for i := range record1{
14
        if record2[i] > 0{
15
            output = append(output, i)
16
        }
17
    }
18

19
    return output
20
}

2.使用字典和集合(代码随想录)

1
func intersection(nums1 []int, nums2 []int) []int {
2
    set:=make(map[int]struct{},0)  // 用map模拟set
3
    res:=make([]int,0)
4
    for _,v:=range nums1{
5
        if _,ok:=set[v];!ok{
6
            set[v]=struct{}{}
7
        }
8
    }
9
    for _,v:=range nums2{
10
        //如果存在于上一个数组中，则加入结果集，并清空该set值
11
        if _,ok:=set[v];ok{
12
            res=append(res,v)
13
            delete(set, v)
14
        }
15
    }
16
    return res
17
}

3.使用数组

和我的方法差不多，但感觉更精准，符合取交集这个要求

1
func intersection(nums1 []int, nums2 []int) []int {
2
    count1 := make([]int, 1001, 1001)
3
    count2 := make([]int, 1001, 1001)
4
    res := make([]int, 0)
5
    for _, v := range nums1 {
6
        count1[v] = 1
7
    }
8
    for _, v := range nums2 {
9
        count2[v] = 1
10
    }
11
    for i := 0; i <= 1000; i++ {
12
        if count1[i] + count2[i] == 2 {
13
            res = append(res, i)
14
        }
15
    }
16
    return res
17
}