Go 博客
Go 映射实战
简介
散列表是计算机科学中最有用的数据结构之一。许多散列表实现具有不同的属性,但通常它们提供快速查找、添加和删除操作。Go 提供了一个内置的 map 类型,它实现了散列表。
声明和初始化
Go map 类型如下所示
map[KeyType]ValueType
其中 KeyType
可以是任何 可比较 的类型(稍后详细介绍),ValueType
可以是任何类型,包括另一个 map!
此变量 m
是一个字符串键到整数值的映射
var m map[string]int
Map 类型是引用类型,如指针或切片,因此上面 m
的值为 nil
;它不指向已初始化的 map。nil map 在读取时表现得像空 map,但尝试写入 nil map 会导致运行时 panic;不要这样做。要初始化 map,请使用内置的 make
函数
m = make(map[string]int)
make
函数分配并初始化一个散列表数据结构,并返回一个指向它的 map 值。该数据结构的具体细节是运行时的实现细节,不受语言本身的规定。在本文中,我们将重点关注 map 的使用,而不是它们的实现。
使用 map
Go 提供了熟悉的语法来操作 map。此语句将键 "route"
设置为值 66
m["route"] = 66
此语句检索存储在键 "route"
下的值,并将其分配给一个新的变量 i
i := m["route"]
如果请求的键不存在,我们将获得值类型的零值。在本例中,值类型为 int
,因此零值为 0
j := m["root"]
// j == 0
内置的 len
函数返回 map 中项目的数量
n := len(m)
内置的 delete
函数从 map 中删除一个条目
delete(m, "route")
delete
函数不返回任何内容,如果指定的键不存在,则不会执行任何操作。
双值赋值测试键是否存在
i, ok := m["route"]
在此语句中,第一个值 (i
) 被分配存储在键 "route"
下的值。如果该键不存在,则 i
为值类型的零值 (0
)。第二个值 (ok
) 是一个 bool
值,如果键存在于 map 中,则为 true
,否则为 false
。
要测试键是否存在而不检索值,请使用下划线代替第一个值
_, ok := m["route"]
要迭代 map 的内容,请使用 range
关键字
for key, value := range m {
fmt.Println("Key:", key, "Value:", value)
}
要使用一些数据初始化 map,请使用 map 字面量
commits := map[string]int{
"rsc": 3711,
"r": 2138,
"gri": 1908,
"adg": 912,
}
相同的语法可用于初始化空 map,这在功能上等同于使用 make
函数
m = map[string]int{}
利用零值
当键不存在时,map 检索产生零值可能很方便。
例如,布尔值的 map 可用作类似集合的数据结构(回想一下,布尔类型的零值为 false)。此示例遍历 Nodes
的链表并打印其值。它使用 Node
指针的 map 来检测列表中的循环。
type Node struct { Next *Node Value interface{} } var first *Node visited := make(map[*Node]bool) for n := first; n != nil; n = n.Next { if visited[n] { fmt.Println("cycle detected") break } visited[n] = true fmt.Println(n.Value) }
表达式 visited[n]
如果 n
已被访问则为 true
,否则如果 n
不存在则为 false
。无需使用双值形式来测试 n
是否存在于 map 中;零值默认值会为我们做到这一点。
另一个有用的零值实例是切片的 map。追加到 nil 切片只会分配一个新的切片,因此将值追加到切片的 map 只需一行代码;无需检查键是否存在。在以下示例中,切片 people 使用 Person
值填充。每个 Person
都有一个 Name
和一个 Likes 切片。此示例创建一个 map,将每个 like 与喜欢它的用户切片相关联。
type Person struct { Name string Likes []string } var people []*Person likes := make(map[string][]*Person) for _, p := range people { for _, l := range p.Likes { likes[l] = append(likes[l], p) } }
打印喜欢奶酪的用户列表
for _, p := range likes["cheese"] { fmt.Println(p.Name, "likes cheese.") }
打印喜欢培根的用户数量
fmt.Println(len(likes["bacon"]), "people like bacon.")
请注意,由于 range 和 len 都将 nil 切片视为零长度切片,因此即使没有人喜欢奶酪或培根(无论多么不可能),最后两个示例也能正常工作。
键类型
如前所述,map 键可以是任何可比较的类型。语言规范 精确地定义了这一点,但简而言之,可比较的类型是布尔型、数值型、字符串型、指针型、通道型和接口类型,以及仅包含这些类型的结构体或数组。值得注意的是,列表中没有切片、map 和函数;这些类型不能使用 ==
进行比较,也不能用作 map 键。
字符串、整数和其他基本类型可用作 map 键是很明显的,但也许意想不到的是结构体键。结构体可用于根据多个维度对数据进行键控。例如,此 map of maps 可用于根据国家/地区统计网页点击量
hits := make(map[string]map[string]int)
这是一个字符串到(字符串到整数的 map)的 map。外部 map 的每个键都是网页的路径,它有自己的内部 map。每个内部 map 键都是一个两位国家/地区代码。此表达式检索澳大利亚人加载文档页面的次数
n := hits["/doc/"]["au"]
不幸的是,这种方法在添加数据时变得笨拙,因为对于任何给定的外部键,您都必须检查内部 map 是否存在,如果不存在则创建它
func add(m map[string]map[string]int, path, country string) {
mm, ok := m[path]
if !ok {
mm = make(map[string]int)
m[path] = mm
}
mm[country]++
}
add(hits, "/doc/", "au")
另一方面,使用单个 map 和结构体键的设计消除了所有这些复杂性
type Key struct {
Path, Country string
}
hits := make(map[Key]int)
当越南人访问主页时,递增(并可能创建)相应的计数器只需一行代码
hits[Key{"/", "vn"}]++
查看有多少瑞士人阅读了规范也很简单
n := hits[Key{"/ref/spec", "ch"}]
并发
map 不安全用于并发使用:当您同时读取和写入它们时,结果是不确定的。如果您需要从并发执行的 goroutine 读取和写入 map,则必须通过某种同步机制来协调访问。一种常见的保护 map 的方法是使用 sync.RWMutex。
此语句声明一个 counter
变量,它是一个匿名结构体,包含一个 map 和一个嵌入的 sync.RWMutex
。
var counter = struct{
sync.RWMutex
m map[string]int
}{m: make(map[string]int)}
要从计数器读取,请获取读锁
counter.RLock()
n := counter.m["some_key"]
counter.RUnlock()
fmt.Println("some_key:", n)
要写入计数器,请获取写锁
counter.Lock()
counter.m["some_key"]++
counter.Unlock()
迭代顺序
使用 range 循环迭代 map 时,迭代顺序未指定,并且不保证在每次迭代中都相同。如果您需要稳定的迭代顺序,则必须维护一个单独的数据结构来指定该顺序。此示例使用键的单独排序切片按键顺序打印 map[int]string
import "sort"
var m map[int]string
var keys []int
for k := range m {
keys = append(keys, k)
}
sort.Ints(keys)
for _, k := range keys {
fmt.Println("Key:", k, "Value:", m[k])
}
下一篇文章:参加 Go 聚会
上一篇文章:go fmt 您的代码
博客索引