Golang 是否在数组中|极客笔记

Golang 是否在数组中

在很多编程语言中，要判断一个元素是否在数组中，常常需要遍历整个数组来进行比较。但在Golang中，由于其内置的map类型的特性，我们可以更高效地判断一个元素是否在数组中。

使用map来实现是否在数组中

在Golang中，map是一种由key-value对组成的集合，其中每个key都是唯一的。我们可以利用map的这个特性来判断某个元素是否在一个数组中。

下面通过一个示例代码来说明如何用map来实现判断元素是否在数组中：

package main

import (
    "fmt"
)

func main() {
    arr := []int{1, 2, 3, 4, 5}

    // 使用map来存储数组中的元素
    mapping := make(map[int]bool)
    for _, v := range arr {
        mapping[v] = true
    }

    // 判断元素是否在数组中
    num := 3
    if _, ok := mapping[num]; ok {
        fmt.Println(num, "is in the array")
    } else {
        fmt.Println(num, "is not in the array")
    }
}

在上面的代码中，首先我们定义了一个数组arr，然后利用一个名为mapping的map来存储数组中的元素。接着我们判断元素3是否在数组中，通过map的特性，我们可以很容易地进行判断。

运行以上代码，输出如下：

3 is in the array

使用slice来实现是否在数组中

除了使用map来实现是否在数组中，我们还可以利用slice的contains方法。

下面通过一个示例代码来说明如何用slice的contains方法来判断元素是否在数组中：

package main

import (
    "fmt"
)

func contains(arr []int, num int) bool {
    for _, v := range arr {
        if v == num {
            return true
        }
    }
    return false
}

func main() {
    arr := []int{1, 2, 3, 4, 5}

    // 判断元素是否在数组中
    num := 3
    if contains(arr, num) {
        fmt.Println(num, "is in the array")
    } else {
        fmt.Println(num, "is not in the array")
    }
}

在上面的代码中，我们定义了一个名为contains的函数，用来判断一个元素是否在数组中。然后我们调用这个函数来判断元素3是否在数组中。

运行以上代码，输出如下：