天天看点

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

算法:

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

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

复制

题目1:

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

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

代码实现:

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
}           

复制

执行结果:

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

题目2:

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

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

代码实现:

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 
}           

复制

执行结果:

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