go语言自定义集合类型

1 集合介绍

集合是一个重要的数据结构，其中：集合中的元素不可重复；集合中的元素无序排列

由于go语言中没有集合元素，现在我们来通过map实现一个集合元素，使其具备以下功能：

增加元素：set

删除元素：delete

判断元素是否存在：isExist

查看集合长度：len

 

2 代码实现

package main

import "fmt"

//定义MySet类型，继承map，key是空接口，可以接收任意类型， value是布尔类型
type MySet map[interface{}]bool

//判断元素是否存在
func (m MySet) isExist(a interface{}) bool {
    return m[a]
}

//返回set长度
func (m MySet) len() int {
    return len(m)
}

//设置值set ,key不存在则新增，key存在则重新赋值, 因此集合中的key不重复
func (m MySet) set(a interface{}) {
    m[a] = true   //m.set(1) 调用set方法， m[1] = true
}

//删除值delete
func (m MySet) delete(a interface{}) {
    delete(m, a)
}

//测试代码
func main() {
    //创建一个set
    var m MySet = make(MySet)  //map是引用类型，需要初始化才能使用，相当于 make(map[interface{}]bool)
    //打印set的长度
    fmt.Println(m.len())
    //放入一个值1 
    m.set(1)
    //放入一个相同值
    m.set(1)
    //打印长度，还是1
    fmt.Println(m.len())
    //判断1是否存在
    fmt.Println(m.isExist(1))
    //删除1
    m.delete(1)
    //判断1是否存在
    fmt.Println(m.isExist(1))
}