Go语言实现自定义集合Set：基于Map的完整方案

为什么Go需要自定义集合Set

Go语言的字典（Map）底层基于哈希表实现，其键值唯一且无序的特性与数学中的集合概念高度契合。相较于Java等语言内置的HashSet，Go开发者需手动封装以获取以下能力：

• 元素唯一性：利用Map键不可重复的特性自动过滤重复值。
• 任意类型支持：通过interface{}作为键的类型，使集合能容纳字符串、整数甚至自定义结构体。
• 内存效率：将Map的值类型定义为bool而非存储具体数据，仅占用一个字节，显著降低存储开销。

第1步：定义HashSet结构与初始化

首先在代码包中声明集合底层容器。推荐使用指针接收者以节省拷贝成本。

步骤 1：声明结构体与构造器

入口：新建 hash_set.go 文件。

1. 定义结构体：内部字段使用map[interface{}]bool。
2. 初始化函数：返回*HashSet指针，避免零值为nil导致的运行时恐慌。

type HashSet struct {
	m map[interface{}]bool
}

func NewHashSet() *HashSet {
	return &HashSet{m: make(map[interface{}]bool)}
}

步骤 2：实现基础增删查改

核心在于直接操作m映射表，利用布尔值的存在性判断元素状态。

• Add(e) bool：若m[e]不存在则写入并返回true，否则返回false。
• Remove(e)：调用内置delete()。
• Clear()：重新实例化make(map[interface{}]bool)释放旧对象。
• Contains(e) bool：直接返回m[e]的布尔值。
• Len() int：返回len(m)。

步骤 3：生成可迭代快照 Elements()

由于Map遍历时元素顺序不确定且可能并发修改，直接返回内部Map不安全。应创建切片副本：

1. 预先分配长度为当前len(m)的切片。
2. 遍历Map并将键填入切片。
3. 处理动态增减导致的长度变化：若实际迭代数小于预分配长度，截取尾部nil部分。

func (set *HashSet) Elements() []interface{} {
	snapshot := make([]interface{}, len(set.m))
	i := 0
	for key := range set.m {
		if i < len(snapshot) {
			snapshot[i] = key
		} else {
			snapshot = append(snapshot, key)
		}
		i++
	}
	return snapshot[:i]
}

第2步：实现高级集合运算

基于基础组件，可组合实现符合集合代数逻辑的高级操作。所有运算均返回新创建的*HashSet实例。

交集 (Intersect)

寻找两个集合共有的元素。为优化性能，通常选择较小的集合进行遍历，检查其元素是否存在于另一个集合中。

func (set *HashSet) Intersect(other *HashSet) *HashSet {
	intersectedSet := NewHashSet()
	// 遍历较小子集以提高效率
	if set.Len() < other.Len() {
		for _, v := range set.Elements() {
			if other.Contains(v) {
				intersectedSet.Add(v)
			}
		}
	} else {
		for _, v := range other.Elements() {
			if set.Contains(v) {
				intersectedSet.Add(v)
			}
		}
	}
	return intersectedSet
}

差集 (Difference)

计算set - other，即存在于A但不存在于B的元素。遍历set的所有元素，排除掉被other.Contains命中项即可。

并集 (Union)

合并两集合。创建一个空集，依次将两集合元素加入。Add方法的唯一性检查会自动去重。

对称差集 (SymmetricDifference)

(A-B) ∪ (B-A)，即只出现在其中一个集合的元素。复用已有的Difference和Union方法实现。

第3步：接口抽象 Set Interface

当业务涉及多种集合实现（如线程安全集合或有序集合）时，提取公共行为到接口中能提高代码泛化能力。

定义通用接口

将上述基础方法与字符串表示、同类比较方法抽象为Set接口：

type Set interface {
	Add(e interface{}) bool
	Remove(e interface{})
	Clear()
	Contains(e interface{}) bool
	Len() int
	Same(other Set) bool
	Elements() []interface{}
	String() string
}

改造现有方法

需调整Same方法和高级运算函数的签名，使其不再依赖具体的*HashSet类型，而是接受Set接口：

1. Same(other Set)：比较时不再需要断言类型，仅需验证长度与包含关系。
2. IsSuperset(one, other Set)：改为包级函数，遍历参数接口的元素进行判定。

通过以上步骤，即可获得一套具备高扩展性、无第三方依赖的Go语言集合工具类。以上就是Go语言之自定义集合Set的详细内容，更多关于Go语言进阶的资料请关注本站其它相关文章！