算法:
求数组的交集,利用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
}
复制
执行结果: