在 go 语言中,求数组交集的方法有:使用内置函数 intersect,适用于已排序数组。使用 map,适用于大型或元素不多的数组。自定义排序和二分搜索,适用于非常大的数组。具体选择哪种方法取决于数组大小和元素分布。
解密 Go 语言数组求交集的技巧
在 Go 语言中,数组是存储同类型元素的有序集合。对数组执行交集操作可以获得两个或多个数组中同时存在的元素。以下介绍了不同场景下求交集的几种实用技巧。
内置函数 Intersect
Go 语言提供了 sort
包中的 Intersect
函数,可直接求取两个已排序数组的交集。该函数接收两个已排序数组作为参数,并返回一个包含交集元素的新数组。
package main
import (
"fmt"
"sort"
)
func main() {
arr1 := []int{1, 3, 5, 7, 9}
arr2 := []int{2, 4, 6, 8, 9}
sort.Ints(arr1)
sort.Ints(arr2)
res := sort.Intersect(arr1, arr2)
fmt.Println(res) // []9
}
对于较小的数组,Intersect
函数是高效且简洁的解决方案。
使用 map
对于大型或元素不多的数组,使用 map
进行求交集也是一种有效的方法。通过将一个数组作为 map
的键,可以快速检查另一个数组中的元素是否也在 map
中。
package main
import "fmt"
func main() {
arr1 := []int{1, 3, 5, 7, 9}
arr2 := []int{2, 4, 6, 8, 9}
m := make(map[int]bool)
for _, v := range arr1 {
m[v] = true
}
var res []int
for _, v := range arr2 {
if m[v] {
res = append(res, v)
}
}
fmt.Println(res) // []9
}
自定义排序和二分搜索
对于非常大的数组,使用自定义排序和二分搜索算法求交集可以实现更好的性能。首先对两个数组进行排序,然后遍历其中一个数组并对另一个数组进行二分搜索,以查找匹配的元素。
package main
import (
"fmt"
"sort"
)
func main() {
arr1 := []int{1, 3, 5, 7, 9, 11, 13, 15}
arr2 := []int{2, 4, 6, 8, 9, 10, 12, 14, 16}
sort.Ints(arr1)
sort.Ints(arr2)
res := intersection(arr1, arr2)
fmt.Println(res) // []9
}
func intersection(a, b []int) []int {
var res []int
for _, v := range a {
idx := sort.SearchInts(b, v)
if idx >= 0 && b[idx] == v {
res = append(res, v)
}
}
return res
}
根据数组大小和元素分布,选择最合适的求交集技巧可以优化代码性能。