解密 Golang 陣列求交集的技巧

在 Go 語言中，求數組交集的方法有：使用內建函數 Intersect，適用於已排序數組。使用 map，適用於大型或元素不多的陣列。自訂排序和二分搜索，適用於非常大的數組。具體選擇哪種方法取決於數組大小和元素分佈。

解密Go 語言陣列求交集的技巧

在Go 語言中，陣列是儲存同類型元素的有序集合。對數組執行交集操作可以獲得兩個或多個數組中同時存在的元素。以下介紹了不同場景下求交集的幾種實用技巧。

內建函數Intersect

Go 語言提供了sort 套件中的Intersect 函數，可直接求取兩個已排序數組的交集。此函數接收兩個已排序數組作為參數，並傳回一個包含交集元素的新數組。

package main

import (
    "fmt"
    "sort"
)

func main() {
    arr1 := []int{1, 3, 5, 7, 9}
    arr2 := []int{2, 4, 6, 8, 9}

    sort.Ints(arr1)
    sort.Ints(arr2)

    res := sort.Intersect(arr1, arr2)
    fmt.Println(res) // []9
}

對於較小的數組，Intersect 函數是高效且簡潔的解決方案。

使用map

對於大型或元素不多的數組，使用map 進行求交集也是一種有效的方法。透過將一個陣列作為 map 的鍵，可以快速檢查另一個陣列中的元素是否也在 map 中。

package main

import "fmt"

func main() {
    arr1 := []int{1, 3, 5, 7, 9}
    arr2 := []int{2, 4, 6, 8, 9}

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

    var res []int
    for _, v := range arr2 {
        if m[v] {
            res = append(res, v)
        }
    }

    fmt.Println(res) // []9
}

自訂排序和二分搜尋

對於非常大的數組，使用自訂排序和二分搜尋演算法求交集可以實現更好的效能。首先對兩個數組進行排序，然後遍歷其中一個數組並對另一個數組進行二分搜索，以查找匹配的元素。

package main

import (
    "fmt"
    "sort"
)

func main() {
    arr1 := []int{1, 3, 5, 7, 9, 11, 13, 15}
    arr2 := []int{2, 4, 6, 8, 9, 10, 12, 14, 16}

    sort.Ints(arr1)
    sort.Ints(arr2)

    res := intersection(arr1, arr2)
    fmt.Println(res) // []9
}

func intersection(a, b []int) []int {
    var res []int
    for _, v := range a {
        idx := sort.SearchInts(b, v)
        if idx >= 0 && b[idx] == v {
            res = append(res, v)
        }
    }
    return res
}

根據陣列大小和元素分佈，選擇最合適的求交集技巧可以最佳化程式碼效能。

以上是解密 Golang 陣列求交集的技巧的詳細內容。更多資訊請關注PHP中文網其他相關文章！