gopl-zh.github.com/ch6/ch6-05.md

114 lines
6.5 KiB
Markdown
Raw Normal View History

2016-02-15 03:06:34 +00:00
## 6.5. 示例: Bit数组
2015-12-09 07:45:11 +00:00
2016-02-15 03:06:34 +00:00
Go语言里的集合一般会用map[T]bool这种形式来表示T代表元素类型。集合用map类型来表示虽然非常灵活但我们可以以一种更好的形式来表示它。例如在数据流分析领域集合元素通常是一个非负整数集合会包含很多元素并且集合会经常进行并集、交集操作这种情况下bit数组会比map表现更加理想。(译注这里再补充一个例子比如我们执行一个http下载任务把文件按照16kb一块划分为很多块需要有一个全局变量来标识哪些块下载完成了这种时候也需要用到bit数组)
2015-12-09 07:45:11 +00:00
2016-02-15 03:06:34 +00:00
一个bit数组通常会用一个无符号数或者称之为“字”的slice或者来表示每一个元素的每一位都表示集合里的一个值。当集合的第i位被设置时我们才说这个集合包含元素i。下面的这个程序展示了一个简单的bit数组类型并且实现了三个函数来对这个bit数组来进行操作
2015-12-09 07:45:11 +00:00
2016-01-21 02:08:07 +00:00
<u><i>gopl.io/ch6/intset</i></u>
2015-12-09 07:45:11 +00:00
```go
// An IntSet is a set of small non-negative integers.
// Its zero value represents the empty set.
type IntSet struct {
2016-01-21 02:08:07 +00:00
words []uint64
2015-12-09 07:45:11 +00:00
}
// Has reports whether the set contains the non-negative value x.
func (s *IntSet) Has(x int) bool {
2016-01-21 02:08:07 +00:00
word, bit := x/64, uint(x%64)
return word < len(s.words) && s.words[word]&(1<<bit) != 0
2015-12-09 07:45:11 +00:00
}
// Add adds the non-negative value x to the set.
func (s *IntSet) Add(x int) {
2016-01-21 02:08:07 +00:00
word, bit := x/64, uint(x%64)
for word >= len(s.words) {
s.words = append(s.words, 0)
}
s.words[word] |= 1 << bit
2015-12-09 07:45:11 +00:00
}
// UnionWith sets s to the union of s and t.
func (s *IntSet) UnionWith(t *IntSet) {
2016-01-21 02:08:07 +00:00
for i, tword := range t.words {
if i < len(s.words) {
s.words[i] |= tword
} else {
s.words = append(s.words, tword)
}
}
2015-12-09 07:45:11 +00:00
}
```
2016-02-15 03:06:34 +00:00
因为每一个字都有64个二进制位所以为了定位x的bit位我们用了x/64的商作为字的下标并且用x%64得到的值作为这个字内的bit的所在位置。UnionWith这个方法里用到了bit位的“或”逻辑操作符号|来一次完成64个元素的或计算。(在练习6.5中我们还会程序用到这个64位字的例子。)
2015-12-09 07:45:11 +00:00
2016-02-15 03:06:34 +00:00
当前这个实现还缺少了很多必要的特性我们把其中一些作为练习题列在本小节之后。但是有一个方法如果缺失的话我们的bit数组可能会比较难混将IntSet作为一个字符串来打印。这里我们来实现它让我们来给上面的例子添加一个String方法类似2.5节中做的那样:
2015-12-09 07:45:11 +00:00
```go
// String returns the set as a string of the form "{1 2 3}".
func (s *IntSet) String() string {
2016-01-21 02:08:07 +00:00
var buf bytes.Buffer
buf.WriteByte('{')
for i, word := range s.words {
if word == 0 {
continue
}
for j := 0; j < 64; j++ {
if word&(1<<uint(j)) != 0 {
if buf.Len() > len("{") {
buf.WriteByte('}')
}
fmt.Fprintf(&buf, "%d", 64*i+j)
2016-01-21 02:08:07 +00:00
}
}
}
buf.WriteByte('}')
return buf.String()
2015-12-09 07:45:11 +00:00
}
```
2016-02-15 03:06:34 +00:00
这里留意一下String方法是不是和3.5.4节中的intsToString方法很相似bytes.Buffer在String方法里经常这么用。当你为一个复杂的类型定义了一个String方法时fmt包就会特殊对待这种类型的值这样可以让这些类型在打印的时候看起来更加友好而不是直接打印其原始的值。fmt会直接调用用户定义的String方法。这种机制依赖于接口和类型断言在第7章中我们会详细介绍。
2015-12-09 07:45:11 +00:00
2016-02-15 03:06:34 +00:00
现在我们就可以在实战中直接用上面定义好的IntSet了
2016-01-02 13:17:21 +00:00
2015-12-09 07:45:11 +00:00
```go
var x, y IntSet
x.Add(1)
x.Add(144)
x.Add(9)
fmt.Println(x.String()) // "{1 9 144}"
y.Add(9)
y.Add(42)
fmt.Println(y.String()) // "{9 42}"
x.UnionWith(&y)
fmt.Println(x.String()) // "{1 9 42 144}"
fmt.Println(x.Has(9), x.Has(123)) // "true false"
```
2016-02-15 03:06:34 +00:00
这里要注意我们声明的String和Has两个方法都是以指针类型*IntSet来作为接收器的但实际上对于这两个类型来说把接收器声明为指针类型也没什么必要。不过另外两个函数就不是这样了因为另外两个函数操作的是s.words对象如果你不把接收器声明为指针对象那么实际操作的是拷贝对象而不是原来的那个对象。因此因为我们的String方法定义在IntSet指针上所以当我们的变量是IntSet类型而不是IntSet指针时可能会有下面这样让人意外的情况
2016-01-02 13:17:21 +00:00
2015-12-09 07:45:11 +00:00
```go
fmt.Println(&x) // "{1 9 42 144}"
fmt.Println(x.String()) // "{1 9 42 144}"
fmt.Println(x) // "{[4398046511618 0 65536]}"
```
2016-01-02 13:17:21 +00:00
2016-02-15 03:06:34 +00:00
在第一个Println中我们打印一个*IntSet的指针这个类型的指针确实有自定义的String方法。第二Println我们直接调用了x变量的String()方法这种情况下编译器会隐式地在x前插入&操作符这样相当远我们还是调用的IntSet指针的String方法。在第三个Println中因为IntSet类型没有String方法所以Println方法会直接以原始的方式理解并打印。所以在这种情况下&符号是不能忘的。在我们这种场景下你把String方法绑定到IntSet对象上而不是IntSet指针上可能会更合适一些不过这也需要具体问题具体分析。
2015-12-09 07:45:11 +00:00
2016-02-15 03:06:34 +00:00
练习6.1: 为bit数组实现下面这些方法
2016-01-02 13:17:21 +00:00
2015-12-09 07:45:11 +00:00
```go
func (*IntSet) Len() int // return the number of elements
func (*IntSet) Remove(x int) // remove x from the set
func (*IntSet) Clear() // remove all elements from the set
func (*IntSet) Copy() *IntSet // return a copy of the set
```
2016-02-15 03:06:34 +00:00
**练习 6.2** 定义一个变参方法(*IntSet).AddAll(...int)这个方法可以为一组IntSet值求和比如s.AddAll(1,2,3)。
2015-12-09 07:45:11 +00:00
2016-02-15 03:06:34 +00:00
**练习 6.3** (*IntSet).UnionWith会用|操作符计算两个集合的交集我们再为IntSet实现另外的几个函数IntersectWith(交集元素在A集合B集合均出现),DifferenceWith(差集元素出现在A集合未出现在B集合),SymmetricDifference(并差集元素出现在A但没有出现在B或者出现在B没有出现在A)。
练习6.4: 实现一个Elems方法返回集合中的所有元素用于做一些range之类的遍历操作。
2015-12-09 07:45:11 +00:00
2016-02-15 03:06:34 +00:00
**练习 6.5** 我们这章定义的IntSet里的每个字都是用的uint64类型但是64位的数值可能在32位的平台上不高效。修改程序使其使用uint类型这种类型对于32位平台来说更合适。当然了这里我们可以不用简单粗暴地除64可以定义一个常量来决定是用32还是64这里你可能会用到平台的自动判断的一个智能表达式32 << (^uint(0) >> 63)