go 计算map 差集、并集、交集

原文链接：> https://www.zhoubotong.site/post/95.html

某些业务场景可能需要实现两个或者多个map 的交差并集，一下举个简单的demo, 思路其实并不复杂：

1. package main
3. import "fmt"
5. // 并集
6. func union(m1, m2 map[string]int) map[string]int {
7.     result := make(map[string]int) // 思路：先把其中一个map 放到新的对象中，把m2中key不存在于本对象中合并即可
8.     for k, v := range m1 {
9.         result[k] = v
10.     }
11.     for k, v := range m2 {
12.         if _, ok := result[k]; !ok {
13.             result[k] = v
14.         }
15.     }
16.     return result
17. }
19. // 交集 两个map中都存在
20. func intersection(m1, m2 map[string]int) map[string]int {
21.     result := make(map[string]int) // 思路：优先遍历m1，判断m2中是否存在相同key,存在则存入新对象
22.     for k, v := range m1 {
23.         if _, ok := m2[k]; ok {
24.             result[k] = v
25.         }
26.     }
27.     return result
28. }
30. // 差集 包括第一个map中存在的键值对，但是不包括第二个map中的键值对
31. func difference(m1, m2 map[string]int) map[string]int {
32.     result := make(map[string]int) // 思路：遍历m1, 判断m2中不存在相同key的则存入新对象，即为差集
33.     for k, v := range m1 {
34.         if _, ok := m2[k]; !ok {
35.             result[k] = v
36.         }
37.     }
38.     return result
39. }
40. func main() {
41.     m1 := make(map[string]int)
42.     m1["a"] = 11
43.     m1["b"] = 12
44.     m1["c"] = 13
45.     m2 := make(map[string]int)
46.     m2["a"] = 11
47.     m2["c"] = 13
48.     fmt.Println("并集：", union(m1, m2))
49.     fmt.Println("交集：", intersection(m1, m2))
50.     fmt.Println("差集：", difference(m1, m2))
51. }

复制代码

输出：

1. [Running] go run "/media/uos/G/web/demo/main.go"
2. 并集： map[a:11 b:12 c:13]
3. 交集： map[a:11 c:13]
4. 差集： map[b:12]

复制代码

代码中有简单说明，不难看出实现过程其实比较简单。

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！