Go语言圣经-指针对象的方法-bit数组习题2详解编程语言

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