Go语言中高效检查字符串切片是否包含特定值

package main

import "fmt"

// ContainsString 检查字符串切片是否包含特定值
func ContainsString(value string, list []string) bool {
    for _, v := range list {
        if v == value {
            return true
        }
    }
    return false
}

func main() {
    list := []string{"a", "b", "x", "golang"}
    fmt.Println(ContainsString("b", list))     // true
    fmt.Println(ContainsString("z", list))     // false
    fmt.Println(ContainsString("golang", list)) // true
}

package main

import "fmt"

// BuildStringSet 从字符串切片构建一个字符串集合（map）
func BuildStringSet(list []string) map[string]bool {
    set := make(map[string]bool, len(list)) // 预分配容量以优化性能
    for _, v := range list {
        set[v] = true
    }
    return set
}

func main() {
    list := []string{"apple", "banana", "cherry", "date"}

    // 第一次构建，O(n) 时间复杂度
    stringSet := BuildStringSet(list)

    // 后续查询，平均 O(1) 时间复杂度
    fmt.Println(stringSet["banana"]) // true
    fmt.Println(stringSet["grape"])  // false
    fmt.Println(stringSet["apple"])  // true
}

package main

import (
    "fmt"
    "sort"
)

// ContainsStringSorted 检查已排序的字符串切片是否包含特定值
func ContainsStringSorted(value string, sortedList []string) bool {
    // sort.SearchStrings 返回目标值可能插入的位置索引
    // 如果目标值存在，它会返回其索引
    // 如果目标值不存在，它会返回一个大于所有现有元素的索引
    i := sort.SearchStrings(sortedList, value)

    // 检查索引是否在切片范围内，并且该位置的元素是否与目标值匹配
    return i < len(sortedList) && sortedList[i] == value
}

func main() {
    list := []string{"zebra", "apple", "banana", "cherry", "date"}

    // 第一次排序，O(n log n) 时间复杂度
    sort.Strings(list) // 原地排序
    fmt.Println("Sorted list:", list) // 输出: [apple banana cherry date zebra]

    // 后续查询，O(log n) 时间复杂度
    fmt.Println(ContainsStringSorted("banana", list)) // true
    fmt.Println(ContainsStringSorted("grape", list))  // false
    fmt.Println(ContainsStringSorted("zebra", list))  // true
}