golang 数组包含|极客笔记

golang 数组包含

在Go语言中，数组是一种固定长度且同一类型元素的序列。本文将详细介绍如何在Go语言中判断一个数组是否包含另一个数组。

方法一：遍历比较

最简单的方法是遍历数组1中的每个元素，看它是否同时存在于数组2中。以下是示例代码：

package main

import (
    "fmt"
)

func contains(arr1 []int, arr2 []int) bool {
    for _, v1 := range arr1 {
        found := false
        for _, v2 := range arr2 {
            if v1 == v2 {
                found = true
                break
            }
        }
        if !found {
            return false
        }
    }
    return true
}

func main() {
    arr1 := []int{1, 2, 3, 4, 5}
    arr2 := []int{2, 4, 6}
    fmt.Println(contains(arr1, arr2)) // Output: false

    arr3 := []int{1, 2, 3}
    arr4 := []int{2, 3}
    fmt.Println(contains(arr3, arr4)) // Output: true
}

在上面的示例中，我们定义了一个contains函数，该函数使用两个for循环来比较数组1中的元素是否都包含在数组2中。如果存在一个元素不包含在数组2中，则返回false，否则返回true。通过main函数我们可以看到两个示例，分别判断数组1和数组2是否包含，输出正确。

方法二：利用Map实现

另一种方法是利用Map来实现。我们遍历数组2，将数组2的每个元素作为键存储在一个Map中。接着遍历数组1，检查数组1的每个元素是否在Map中存在。如果数组1中的所有元素都在Map中存在，则说明数组1包含数组2。以下是示例代码：

package main

import "fmt"

func contains(arr1 []int, arr2 []int) bool {
    m := make(map[int]bool)
    for _, v := range arr2 {
        m[v] = true
    }

    for _, v := range arr1 {
        if _, ok := m[v]; !ok {
            return false
        }
    }

    return true
}

func main() {
    arr1 := []int{1, 2, 3, 4, 5}
    arr2 := []int{2, 4, 6}
    fmt.Println(contains(arr1, arr2)) // Output: false

    arr3 := []int{1, 2, 3}
    arr4 := []int{2, 3}
    fmt.Println(contains(arr3, arr4)) // Output: true
}

在上面的示例中，我们同样定义了一个contains函数，该函数使用Map来存储数组2中的每个元素，然后遍历数组1，检查数组1中的每个元素是否在Map中存在。通过main函数我们可以看到结果符合预期。

方法三：使用库函数

Go语言提供了一些库函数可以实现判断数组包含的功能。例如sort.Ints函数可以用于对整型数组进行排序，在排序后可以使用search.Ints函数判断是否包含。以下是示例代码：

package main

import (
    "fmt"
    "sort"
)

func contains(arr1 []int, arr2 []int) bool {
    sort.Ints(arr1)
    sort.Ints(arr2)

    for _, v := range arr2 {
        if !sort.IntsAreSorted(arr1) {
            return false
        }
        index := sort.SearchInts(arr1, v)
        if index == len(arr1) || arr1[index] != v {
            return false
        }
    }
    return true
}

func main() {
    arr1 := []int{1, 2, 3, 4, 5}
    arr2 := []int{2, 4, 6}
    fmt.Println(contains(arr1, arr2)) // Output: false

    arr3 := []int{1, 2, 3}
    arr4 := []int{2, 3}
    fmt.Println(contains(arr3, arr4)) // Output: true
}

在上面的示例中，我们使用sort.Ints函数对数组1和数组2进行排序，然后使用sort.SearchInts函数在排序后的数组1中查找元素是否存在。通过main函数我们可以看到结果符合预期。

通过以上三种方法，我们可以在Go语言中判断一个数组是否包含另一个数组。可以根据实际情况选择最适合的方法。