登录
推荐 文章 Go 技术 课程 下载 专题 AI
首页 >  Golang >  Go教程

Go bitset 怎样对两组标记计算对称差

来源:17golang原创

时间:2026-10-31 09:08:48 212浏览 收藏

两组标记要找“只属于其中一组”的编号时,不需要自己逐字做异或。github.com/bits-and-blooms/bitset 已经提供了 SymmetricDifference:它返回一个新 BitSet,结果中只保留 A 或 B 设置、但没有同时设置的位。如果允许修改其中一组,则可用 InPlaceSymmetricDifference 直接复用接收者。

官方地址:https://pkg.go.dev/github.com/bits-and-blooms/bitset

先记住三个判断
  • 对称差等价于“并集减去交集”,相同编号在两组中都出现时会被排除。
  • SymmetricDifference 生成新集合;InPlaceSymmetricDifference 会改写接收者。
  • 编号越大,BitSet 的长度上界越大;如果数据极其稀疏,应重新评估普通 bitset 是否合适。

先把“对称差”与普通差集分开

假设 A 表示本次活动用户,B 表示上次活动用户。A 是 1、3、5、8,B 是 3、4、8、10。普通差集 A-B 只会得到 1、5;对称差则要把两边独有的编号都留下,因此结果是 1、4、5、10。编号 3 和 8 同时存在于两组,必须被排除。

Go bitset 两组标记 A 和 B 经过交集与并集关系后保留单边编号的静态结构说明图
图1:输入关系说明图,展示两组标记、共同编号和对称差结果之间的静态边界。

用 SymmetricDifference 保留两组输入

默认推荐非原地方法,因为它把“计算结果”和“原始数据”分开。这个选择适合后面还要分别统计活动用户与历史用户的场景,也能避免把共享的 *BitSet 意外改掉。

package main

import (
    "fmt"

    "github.com/bits-and-blooms/bitset"
)

func main() {
    current := bitset.New(16)
    current.Set(1).Set(3).Set(5).Set(8) // 本次活动的用户编号

    previous := bitset.New(16)
    previous.Set(3).Set(4).Set(8).Set(10) // 上次活动的用户编号

    changed := current.SymmetricDifference(previous) // 返回只在一边出现的新集合
    fmt.Println("差异数量:", changed.Count())             // 结果应包含 1、4、5、10

    for i, ok := changed.NextSet(0); ok; i, ok = changed.NextSet(i + 1) {
        fmt.Println("差异编号:", i) // 只遍历结果集合中的置位编号
    }
}

这里的重点不是把两组位图逐位对齐,而是让库按集合语义计算。Count 表示结果中的置位数量,NextSet 从给定位置寻找下一个已设置的位;如果项目已经使用 Go 1.23,也可以考虑文档中的 EachSet 迭代方式。

需要节省分配时再用原地版本

当 current 只是临时工作集、计算后不再需要原值时,可以直接在它上面做原地对称差。这样写更短,也明确表达“current 将被消费”。

func symmetricDifferenceInPlace() []uint {
    current := bitset.New(16)
    current.Set(1).Set(3).Set(5).Set(8) // 这个集合允许被改写

    previous := bitset.New(16)
    previous.Set(3).Set(4).Set(8).Set(10) // 对照集合保持不变

    current.InPlaceSymmetricDifference(previous) // 结果直接写回 current
    result := make([]uint, 0, current.Count())
    for i, ok := current.NextSet(0); ok; i, ok = current.NextSet(i + 1) {
        result = append(result, i) // 把差异编号交给后续业务层
    }
    return result
}

原地版本最容易出错的地方是别名:如果其他结构仍持有 current 指针,它们看到的内容也会一起变化。要保留快照时,先对集合调用 Clone,再在副本上调用原地方法。

Go bitset 的非原地与原地对称差 API 分别连接输入集合、结果集合和遍历输出的静态结构说明图
图2:API 选择说明图,区分生成新结果的调用关系与改写接收者的调用关系。

高编号和稀疏数据是实际边界

BitSet 的内存占用与最大访问位编号相关,而不是只与当前置位数量相关。例如只设置编号 10 和 10_000_000,中间的大范围仍会影响位图长度。官方文档还说明 BitSet 不会自动缩小,可按生命周期选择 Shrink 或 Compact。

情况推荐写法原因
还要保留 A、BA.SymmetricDifference(B)结果是独立集合,输入语义清晰
A 只是临时工作集A.InPlaceSymmetricDifference(B)复用 A,减少额外结果对象
编号很大且很稀疏评估压缩位图或其他集合结构普通 bitset 可能为大量空位付出内存

最后可以用三条检查固定实现:相同编号同时出现在两边时结果必须为假;只在一边出现的编号必须保留;原地版本调用后,必须确认接收者是否仍被其他代码共享。这样就能把“计算错了”和“集合被意外改写”区分开。

常见问题

对称差与并集有什么区别?

并集保留两边所有编号;对称差会排除两边共同出现的编号。

怎样只统计变化数量而不遍历结果?

可以先生成对称差后调用 Count;如果只需要基数,写入结果集合再遍历不是必要的。

调用 SymmetricDifference 会修改原集合吗?

按 API 语义应把它当作返回新结果的方法;需要明确修改接收者时,使用名称中带 InPlace 的版本,并在代码审查中标注所有权。

声明:本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>