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

时间:2022-09-07 19:53:57

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