练习 6.3: (*IntSet).UnionWith会用|操作符计算两个集合的交集,我们再为IntSet实现另外的几个函数IntersectWith(交集:元素在A集合B集合均出现),DifferenceWith(差集:元素出现在A集合,未出现在B集合),SymmetricDifference(并差集:元素出现在A但没有出现在B,或者出现在B没有出现在A)。
*练习6.4: 实现一个Elems方法,返回集合中的所有元素,用于做一些range之类的遍历操作。
练习 6.2: 定义一个变参方法(*IntSet).AddAll(...int),这个方法可以添加一组IntSet,比如s.AddAll(1,2,3)。
代码语言:javascript复制/*
练习 6.3: (*IntSet).UnionWith会用|操作符计算两个集合的交集,我们再为IntSet实现另外的几个函数IntersectWith(交集:元素在A集合B集合均出现),DifferenceWith(差集:元素出现在A集合,未出现在B集合),Sym
metricDifference(并差集:元素出现在A但没有出现在B,或者出现在B没有出现在A)。
*/
//交集:元素在A集合B集合均出现
func (s *IntSet) IntersectWith(t *IntSet) IntSet {
var result IntSet
for i, word := range s.words {
if i >= len(t.words) {
break
}
result.words = append(result.words, word&t.words[i])
}
return result
}
//差集:元素出现在A集合,未出现在B集合
func (s *IntSet) DifferenceWith(t *IntSet) IntSet {
var result IntSet
for i, word := range s.words {
if i >= len(t.words) {
result.words = append(result.words, word)
continue
}
result.words = append(result.words, word&(word^t.words[i]))
}
return result
}
//并差集:元素出现在A但没有出现在B,或者出现在B没有出现在A
func (s *IntSet) SymmetricDifference(t *IntSet) IntSet {
var result IntSet
for i, word := range s.words {
if i >= len(t.words) {
result.words = append(result.words, word)
continue
}
result.words = append(result.words, word^t.words[i])
}
return result
}
//练习6.4: 实现一个Elems方法,返回集合中的所有元素,用于做一些range之类的遍历操作。
func (s *IntSet) Elems() []int {
var result []int
for i, word := range s.words {
for j := 0; j < 64; j {
if word&(1<<uint(j)) != 0 {
result = append(result, 64*i j)
}
}
}
return result
}
//练习 6.2: 定义一个变参方法(*IntSet).AddAll(...int),这个方法可以添加一组IntSet,比如s.AddAll(1,2,3)。
func (s *IntSet) AddAll(elements ...int) {
for _, r := range elements {
s.Add(r)
}
}