跳到主要内容

Go 映射(Map)

介绍​

在Go语言中，**映射(Map)**是一种非常强大的数据结构，用于存储键值对。它类似于其他编程语言中的字典或哈希表。映射允许你通过键快速查找、插入或删除值，这使得它在处理需要快速查找的场景中非常有用。

映射的键可以是任何可比较的类型（如int、string等），而值可以是任意类型。映射的底层实现是一个哈希表，因此它的查找、插入和删除操作的平均时间复杂度为O(1)。

创建和初始化映射​

在Go中，你可以使用make函数来创建一个映射，也可以使用字面量语法来初始化一个映射。

使用make函数创建映射​

go
// 创建一个键为string类型，值为int类型的映射
m := make(map[string]int)

使用字面量初始化映射​

go
// 使用字面量初始化一个映射
m := map[string]int{
    "apple":  5,
    "banana": 3,
    "orange": 2,
}

操作映射​

插入和更新元素​

你可以通过指定键来插入或更新映射中的元素。

go
m := make(map[string]int)
m["apple"] = 5  // 插入键值对
m["banana"] = 3 // 插入键值对
m["apple"] = 7  // 更新键"apple"对应的值

访问元素​

通过键访问映射中的元素。如果键不存在，映射会返回值类型的零值。

go
fmt.Println(m["apple"])  // 输出: 7
fmt.Println(m["grape"])  // 输出: 0 (因为"grape"不存在)

检查键是否存在​

你可以通过第二个返回值来检查键是否存在。

go
value, exists := m["apple"]
if exists {
    fmt.Println("apple exists with value:", value)
} else {
    fmt.Println("apple does not exist")
}

删除元素​

使用delete函数可以从映射中删除一个键值对。

go
delete(m, "banana")
fmt.Println(m)  // 输出: map[apple:7 orange:2]

遍历映射​

你可以使用for range循环来遍历映射中的所有键值对。

go
for key, value := range m {
    fmt.Printf("Key: %s, Value: %d\n", key, value)
}

映射的实际应用场景​

统计单词频率​

假设你有一段文本，想要统计每个单词出现的频率。你可以使用映射来实现这一功能。

go
text := "apple banana apple orange banana apple"
words := strings.Split(text, " ")

wordCount := make(map[string]int)
for _, word := range words {
    wordCount[word]++
}

fmt.Println(wordCount)  // 输出: map[apple:3 banana:2 orange:1]

缓存数据​

映射还可以用于实现简单的缓存系统。例如，你可以将函数的计算结果存储在映射中，以避免重复计算。

go
var cache = make(map[int]int)

func fibonacci(n int) int {
    if n < 2 {
        return n
    }
    if val, exists := cache[n]; exists {
        return val
    }
    cache[n] = fibonacci(n-1) + fibonacci(n-2)
    return cache[n]
}

fmt.Println(fibonacci(10))  // 输出: 55

总结​

映射是Go语言中非常实用的数据结构，适用于需要快速查找、插入和删除的场景。通过本文，你已经学习了如何创建、初始化、操作映射，并了解了映射在实际应用中的一些常见用法。

提示

映射的键必须是可比较的类型，这意味着你不能使用切片、函数或包含切片的结构体作为映射的键。

警告

映射是引用类型，因此在函数间传递映射时，传递的是映射的引用，而不是副本。对映射的修改会影响到所有引用该映射的地方。

附加资源​

练习​

  1. 创建一个映射，存储学生的姓名和成绩，并编写一个函数来查找某个学生的成绩。
  2. 编写一个程序，统计一段文本中每个字符出现的次数。
  3. 使用映射实现一个简单的电话簿，支持添加、删除和查找联系人。

通过完成这些练习，你将更深入地理解映射的使用方法。