当前位置：首页 > 脚本中心 > 其它 > 详细页面

golang中set数据结构的使用示例

时间：2023-03-09来源：系统城装机大师作者：佚名

借助map实现

golang中没有set数据结构，一般是通过map实现，因为map的key值是不能重复的

示例

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24 type empty struct {
}

// golang中的set数据类型
func MakeGolangSet(){
// 首先，空结构体内存消耗为0，其它数据结构如布尔值等均存在内存消耗
var e empty
fmt.Printf("empty struct memory is %v\n", unsafe.Sizeof(e))

fmt.Printf("bool memory is %v\n", unsafe.Sizeof(false))

set := make(map[string]struct{})
// ADD
set["foo"] = e
// DELETE
delete(set, "foo")
// set的规模
size := len(set)
fmt.Printf("set size is %v\n", size)
// 判断是否存在
set["foo"] = e
value, exist := set["foo"]
fmt.Printf("value is %v\n, exist is %v\n", value, exist)
}

算法应用

找两个数组的交集

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 // 两个数组的交集
func SectionWithTwoArray(num1, num2 []int) []int{
set := make(map[int]struct{}, 0)
var res []int
for _, v := range num1 {
if _, ok := set[v]; !ok {
// 去重
set[v] = struct{}{}
}
}
for _, v := range num2 {
if _, ok := set[v]; ok {
res = append(res, v)
// 将此v删掉，避免结果中出现重复数据
delete(set, v)
}
}
return res
}