Go -- Map

声明

1
2
3
4
5
6
7
8
9
10
11
12
13
14
func TestInitMap(t *testing.T) {
m1 := map[int]int{1: 1, 2: 4, 3: 9}
t.Log(m1[2]) // 4
t.Log(len(m1)) // 3

m2 := map[int]int{}
m2[4] = 16
t.Log(len(m2)) // 1

// 10为cap,没有使用len,因为len会初始化为“0”值,但map没法预设“0”值
m3 := make(map[int]int, 10)
t.Log(len(m3)) // 0
//t.Log(cap(m3)) // invalid argument m3 (type map[int]int) for cap
}

元素访问

1
2
3
4
5
6
7
8
9
10
11
12
13
14
func TestAccessNotExistingKey(t *testing.T) {
m1 := map[int]int{}
// 不存在,返回0,避免了其他语言中的空指针异常
t.Log(m1[1]) // 0
m1[2] = 0
// 存在0,返回0,无法区分
t.Log(m1[2]) // 0

if v, ok := m1[3]; ok {
t.Log("Key exists, value=", v)
} else {
t.Log("Key not exists ")
}
}

遍历

1
2
3
4
5
6
7
8
9
func TestTravelMap(t *testing.T) {
m := map[int]int{1: 1, 2: 4, 3: 9}
for k, v := range m {
t.Log(k, v)
// 1 1
// 2 4
// 3 9
}
}

工厂模式

  1. Map的value可以是一个方法
  2. 与Go的Dock type接口方式一起,可以方便地实现单一方法对象的工厂模式
  3. 在Go语言,函数是一等公民
1
2
3
4
5
6
7
func TestMapWithFunValue(t *testing.T) {
m := map[int]func(op int) int{}
m[1] = func(op int) int { return op }
m[2] = func(op int) int { return op * op }
m[3] = func(op int) int { return op * op * op }
t.Log(m[1](2), m[2](2), m[3](2)) // 2 4 8
}

实现Set

  1. Go的内置集合没有Set实现,可以使用map[type]bool
  2. 元素的唯一性
  3. 基本操作:添加元素、判断元素是否存在、删除元素、元素个数
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
func TestMapForSet(t *testing.T) {
mySet := map[int]bool{}
n := 1
mySet[n] = true
if mySet[n] {
t.Logf("%d is exists", n) // 1 is exists
} else {
t.Logf("%d is not exists", n)
}

mySet[2] = true
t.Log(len(mySet)) // 2

delete(mySet, 1)
if mySet[n] {
t.Logf("%d is exists", n)
} else {
t.Logf("%d is not exists", n) // 1 is not exists
}
}
0%