go实现一个set

公众号guangcity

发布于 2021-08-26 10:51:31

4980

发布于 2021-08-26 10:51:31

在golang中没有内置的set，我们可以使用map[interface{}]bool或者map[interface{}]struct{}实现。

为了支持任意类型元素，我们可以使用空接口interface{}，实现set这里使用struct{}

type exists struct{}
type set struct {
 m map[interface{}]exists
}

基本的操作如下：

去调用Add函数添加一堆元素到set中

func NewSet(items ...interface{}) *set {
 s := &set{}
 s.m = make(map[interface{}]exists)
 s.Add(items...)
 return s
}

将元素列表插入到set中

func (s *set) Add(items ...interface{}) {
 for _, item := range items {
  s.m[item] = exists{}
 }
}

删除指定元素

func (s *set) Remove(item interface{}) {
 delete(s.m, item)
}

基本的map查询操作即可

func (s *set) Contains(item interface{}) bool {
 _, ok := s.m[item]
 return ok
}

func (s *set) Contains(item interface{}) bool {
 _, ok := s.m[item]
 return ok
}

func (s *set) Size() int {
 return len(s.m)
}

最后聊一下为啥这里可以用struct{}？

因为你去比较两个struct{}时会发现是相等的。

例如：

s1 := struct{}{}
s2 := struct{}{}
fmt.Printf("s1 address:%p, s2 address:%p\n", &s1, &s2)
fmt.Println(s1 == s2)

此时输出：

s1 address:0x1190370, s2 address:0x1190370
true

可以看到地址一样，比较也相等，空结构体大小多少呢

fmt.Println(unsafe.Sizeof(struct{}{}))

结果为0，精彩呀，不占用内存。放在map构造的set简直无敌，完成了下面两个工作：

上面set使用：

func main() {
 s := NewSet("beijing", "beijing", "shanghai", 101)
 s.Add(false)
}

本节完～

本文参与腾讯云自媒体分享计划，分享自微信公众号。

原始发表：2021-07-30，如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自光城微信公众号，前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体分享计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度