【Golang】判断数组中是否包含指定元素

Amctwd2024-05-12 6:05

本文将介绍如何在Go语言中实现这一功能，并提供一个通用的函数IsContain，用于检查切片中是否包含指定的元素。

实现

下面是一个IsContain函数的实现，它可以检查int和string类型的切片是否包含特定的元素：

go 复制代码

package main
import "fmt"
func IsContain(items interface{}, item interface{}) bool {
	switch items.(type) {
	case []int:
		intArr := items.([]int)
		for _, value := range intArr {
			if value == item.(int) {
				return true
			}
		}
	case []string:
		strArr := items.([]string)
		for _, value := range strArr {
			if value == item.(string) {
				return true
			}
		}
	default:
		return false
	}
	return false
}
func main() {
	intArr := []int{1, 2, 3, 4, 5}
	strArr := []string{"apple", "banana", "cherry"}
	fmt.Println(IsContain(intArr, 3)) // 输出: true
	fmt.Println(IsContain(strArr, "banana")) // 输出: true
	fmt.Println(IsContain(intArr, 6)) // 输出: false
	fmt.Println(IsContain(strArr, "orange")) // 输出: false
}

这个函数通过类型断言来检查items参数的实际类型，并根据类型执行相应的循环和比较操作。如果传入的类型不是int或string切片，函数将返回false。

结论

本文提供了一个Go语言的函数IsContain，用于检查一个切片是否包含特定的元素，适用于int和string类型的切片。

这个函数可以扩展功能，以支持更多的数据类型。