算法篇：利用map求数组交集

灰子学技术

发布于 2020-09-07 07:19:39

72100

代码可运行

文章被收录于专栏：灰子学技术灰子学技术

运行总次数：0

代码可运行

算法：

求数组的交集，利用map的key,value特性会比较简单，步骤如下：

1.先遍历数组1，然后将数组存到map1中
2.遍历数组2，将数组存入map2中，存的过程中需要判断是否存在与map1中，
根据题目要求做去重处理或者不去重处理。
3.将满足条件的元素放到新的数组里面。
（备注：如果是多个数组的话，可以两两求交集，然后依次递减，直到数组个数为1.）

题目1: https://leetcode-cn.com/problems/intersection-of-two-arrays/

代码实现：

func intersection(nums1 []int, nums2 []int) []int {
    var numMap,numMap1 map[int]bool 
    var tar []int
    numMap =make(map[int]bool)
    numMap1 =make(map[int]bool)
    for _,i:=range nums1 { // 遍历数组1排除重复的数字
       if  _,v:=numMap[i];!v {
            numMap[i] = true
       }
    }
    for _,j:=range nums2 {
       if  _,v:=numMap1[j];!v { // 遍历数组2排除重复的数字
           if _,v1:=numMap[j];v1{ // 求数组1和数组2的交集
               numMap1[j] = true
                tar = append(tar,j)
           }
       }
    }
    return tar
}

执行结果：

题目2: https://leetcode-cn.com/problems/intersection-of-two-arrays-ii/

代码实现：

func intersect(nums1 []int, nums2 []int) []int {
    if len(nums1) == 0 || len(nums2) == 0 {
        return nil
    }
    var tar []int
    nMap := make(map[int]int)
    for _,n := range nums1 {
        _, ok := nMap[n]
        if !ok {
            nMap[n] = 1
        } else {
            nMap[n]++
        }
    }
    for _, m := range nums2 {
        _, ok := nMap[m]
        if ok && nMap[m]>0{
            nMap[m]--
            tar = append(tar,m)
        }
    }
    return tar 
}