Golang slices.CompareFunc
最后修改于 2025 年 4 月 20 日
本教程将介绍如何在 Go 中使用 `slices.CompareFunc` 函数。我们将通过实际示例和自定义比较函数来讲解切片比较操作。
`slices.CompareFunc` 函数使用自定义比较函数对两个切片进行逐元素比较。它是 Go 实验性 slices 包的一部分，提供了灵活的比较功能。
当您需要简单的相等性检查之外的自定义比较逻辑时，此函数非常有用。它根据相应元素的比较结果返回 -1、0 或 1。
基本的 slices.CompareFunc 示例
`slices.CompareFunc` 最简单的用法是比较两个整数切片。我们定义一个返回标准 -1、0、1 排序的比较函数。
package main
import (
"fmt"
"slices"
)
func main() {
s1 := []int{1, 2, 3}
s2 := []int{1, 2, 4}
cmp := slices.CompareFunc(s1, s2, func(a, b int) int {
if a < b {
return -1
}
if a > b {
return 1
}
return 0
})
fmt.Println("Comparison result:", cmp)
}
我们比较了两个相似的整数切片，其中一个元素不同。自定义函数实现了标准的整数比较逻辑。结果是 -1，因为 s1 在字典顺序上小于 s2。
不区分大小写的字符串比较
`slices.CompareFunc` 可以使用自定义逻辑比较字符串切片。此示例执行字符串元素的不区分大小写的比较。
package main
import (
"fmt"
"slices"
"strings"
)
func main() {
s1 := []string{"apple", "Banana", "cherry"}
s2 := []string{"Apple", "banana", "Cherry"}
cmp := slices.CompareFunc(s1, s2, func(a, b string) int {
aLower := strings.ToLower(a)
bLower := strings.ToLower(b)
if aLower < bLower {
return -1
}
if aLower > bLower {
return 1
}
return 0
})
fmt.Println("Case-insensitive comparison:", cmp)
}
比较函数在比较前将字符串转换为小写。这使得比较不区分大小写。结果是 0（相等），因为切片包含相同单词但大小写不同。
比较结构体切片
我们可以将 `slices.CompareFunc` 用于自定义结构体类型。此示例按年龄字段比较 Person 结构体切片。
package main
import (
"fmt"
"slices"
)
type Person struct {
Name string
Age int
}
func main() {
group1 := []Person{
{"Alice", 25},
{"Bob", 30},
}
group2 := []Person{
{"Alice", 25},
{"Bob", 35},
}
cmp := slices.CompareFunc(group1, group2, func(a, b Person) int {
if a.Age < b.Age {
return -1
}
if a.Age > b.Age {
return 1
}
return 0
})
fmt.Println("Person comparison by age:", cmp)
}
比较函数仅考虑 Person 结构体的 Age 字段。结果是 -1，因为 group1 中的 Bob 比 group2 中的年轻。在此比较中忽略了 Name 字段。
比较不同长度的切片
`slices.CompareFunc` 可以处理不同长度的切片。此示例演示了比较不同长度切片时的行为。
package main
import (
"fmt"
"slices"
)
func main() {
s1 := []int{1, 2, 3}
s2 := []int{1, 2, 3, 4}
cmp := slices.CompareFunc(s1, s2, func(a, b int) int {
if a < b {
return -1
}
if a > b {
return 1
}
return 0
})
fmt.Println("Different length comparison:", cmp)
}
当切片长度不同时，较短的切片被认为较小。结果是 -1，因为 s1 比 s2 短，即使它们的共同元素相等。这遵循字典顺序规则。
自定义浮点数比较（带容差）
对于浮点数，我们通常需要带容差的比较。此示例实现了 float64 切片的近似相等性。
package main
import (
"fmt"
"math"
"slices"
)
func main() {
s1 := []float64{1.0, 2.0, 3.0000001}
s2 := []float64{1.0, 2.0, 3.0}
cmp := slices.CompareFunc(s1, s2, func(a, b float64) int {
if math.Abs(a-b) < 0.0001 {
return 0
}
if a < b {
return -1
}
return 1
})
fmt.Println("Float comparison with tolerance:", cmp)
}
比较函数使用一个小的 epsilon 值（0.0001）来确定相等性。在此容差范围内的数字被视为相等。结果是 0，因为使用此容差时 3.0000001 和 3.0 被视为相等。
反向排序比较
我们可以通过反转比较逻辑来实现反向排序。此示例按降序比较切片。
package main
import (
"fmt"
"slices"
)
func main() {
s1 := []int{5, 3, 1}
s2 := []int{5, 2, 1}
cmp := slices.CompareFunc(s1, s2, func(a, b int) int {
if a > b {
return -1
}
if a < b {
return 1
}
return 0
})
fmt.Println("Reverse order comparison:", cmp)
}
比较函数返回反转的结果以实现降序。结果是 -1，因为在反向排序中，s1 中的 3 大于 s2 中的 2。这演示了如何实现自定义排序逻辑。
实际示例：版本号比较
这个实际示例使用自定义逻辑比较版本号切片。版本号按组件进行比较。
package main
import (
"fmt"
"slices"
"strconv"
"strings"
)
func main() {
v1 := []string{"1", "2", "3"}
v2 := []string{"1", "10", "0"}
cmp := slices.CompareFunc(v1, v2, func(a, b string) int {
aNum, _ := strconv.Atoi(a)
bNum, _ := strconv.Atoi(b)
if aNum < bNum {
return -1
}
if aNum > bNum {
return 1
}
return 0
})
fmt.Println("Version comparison result:", cmp)
}
版本字符串在比较前转换为数字。结果是 -1，因为第二个组件中的 2 小于 10。这演示了版本号比较的实际用法。
来源
本教程通过各种场景下使用自定义逻辑比较切片的实际示例，介绍了 Go 中的 `slices.CompareFunc` 函数。