[日常] Go语言圣经-指针对象的方法-bit数组习题2

Posted 2020-10-31 陶士涵的菜地

练习 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)。

/*
练习 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)
        }
}