在Golang中使用泛型实现“slice包含”检查函数

22-03-17 banq

检查一个数组是否包含一个给定的值是一个流行的编程函数,在许多编程语言中都是原生实现的。
在Go中,我们必须自己实现它,但由于新的泛型功能,我们可以编写一个单一的contains()函数,它将适用于任何类型的slice,其值可以被比较。

package main

import "fmt"

func contains[T comparable](elems []T, v T) bool {
    for _, s := range elems {
        if v == s {
            return true
        }
    }
    return false
}

func main() {
    fmt.Println(contains([]string{"a", "b", "c"}, "b"))
    fmt.Println(contains([]int{1, 2, 3}, 2))
    fmt.Println(contains([]int{1, 2, 3}, 10))
}


泛型函数的类型参数在函数名后面的方括号中声明。
在contains()中,我们使用了带有comparable约束的T类型参数。这是一个内置的约束,它描述了任何类型的值可以被比较,也就是说,我们可以对它们使用==和!=运算符。
该函数的主体其实很简单,与非通用版本没有区别。
我们在elems slice上迭代,检查当前值是否是我们要找的值v。这样,我们就得到了一个可以对任何片断类型进行操作的函数。

输出:

true
true
false


 

猜你喜欢