golang练手小项目系列(1)-位向量

栏目: Go · 发布时间: 5年前

内容简介:本系列整理了10个工作量和难度适中的Golang小项目,适合已经掌握Go语法的工程师进一步熟练语法和常用库的用法。有一组非负整数,实现一个位向量类型,能在O(1)时间内完成插入、删除和查找等操作。

本系列整理了10个工作量和难度适中的Golang小项目,适合已经掌握 Go 语法的工程师进一步熟练语法和常用库的用法。

问题描述:

有一组非负整数,实现一个位向量类型,能在O(1)时间内完成插入、删除和查找等操作。

要点:

  • 实现Has(uint)、Add(uint)、Remove(uint)、Clear()、Copy()、String()、AddAll(…uint)、UnionWith()、IntersectWith()、DifferenceWith()、SymmetricDifference()方法。

拓展:

  • 使用uint存储而不是uint32或uint64这样限定字长的类型。

代码实现:

import (
   "bytes"
   "fmt"
)


func (s *IntSet)countBit(n uint)int{
   count := 0

   for n != 0{
      n = n & ( n - 1 )
      count += 1
   }

   return count
}


func (s *IntSet)calWordBit(x int)(word int, bit uint)  {
   word, bit = x / wordSize, uint(x%wordSize)
   return
}



const wordSize = 32 << (^uint(0) >> 63)

// An IntSet is a set of small non-negative integers.
// Its zero value represents the empty set.
type IntSet struct {
   words []uint
}

// Has reports whether the set contains the non-negative value x.
func (s *IntSet) Has(x int) bool {
   word, bit := s.calWordBit(x)
   return word < len(s.words) && s.words[word]&(1<<bit) != 0
}

// Add adds the non-negative value x to the set.
func (s *IntSet) Add(x int) {
   word, bit := s.calWordBit(x)
   for word >= len(s.words) {
      s.words = append(s.words, 0)
   }
   s.words[word] |= 1 << bit
}

func (s *IntSet) AddAll(nums ...int) {
   for _, n := range nums {
      s.Add(n)
   }
}

// UnionWith sets s to the union of s and t.
func (s *IntSet) UnionWith(t *IntSet) {
   for i, tword := range t.words {
      if i < len(s.words) {
         s.words[i] |= tword
      } else {
         s.words = append(s.words, tword)
      }
   }
}

// Set s to the intersection of s and t.
func (s *IntSet) IntersectWith(t *IntSet) {
   for i, tword := range t.words {
      if i < len(s.words) {
         s.words[i] &= tword
      } else {
         s.words = append(s.words, tword)
      }
   }
}

// Set s to the difference of s and t.
func (s *IntSet) DifferenceWith(t *IntSet) {
   for i, tword := range t.words {
      if i < len(s.words) {
         s.words[i] &^= tword
      } else {
         s.words = append(s.words, tword)
      }
   }
}

// Set s to the symmetric difference of s and t.
func (s *IntSet) SymmetricDifference(t *IntSet) {
   for i, tword := range t.words {
      if i < len(s.words) {
         s.words[i] ^= tword
      } else {
         s.words = append(s.words, tword)
      }
   }
}

// return the number of elements
func (s *IntSet) Len() int {
   count := 0
   for _, word := range s.words {
      count += s.countBit(word)
   }
   return count
}

// remove x from the set
func (s *IntSet) Remove(x int) {
   word, bit := s.calWordBit(x)
   s.words[word] &^= 1 << bit
}

// remove all elements from the set
func (s *IntSet) Clear() {
   for i := range s.words {
      s.words[i] = 0
   }
}

// return a copy of the set
func (s *IntSet) Copy() *IntSet {
   new := &IntSet{}
   new.words = make([]uint, len(s.words))
   copy(new.words, s.words)
   return new
}

// String returns the set as a string of the form "{1 2 3}".
func (s *IntSet) String() string {
   var buf bytes.Buffer
   buf.WriteByte('{')
   for i, word := range s.words {
      if word == 0 {
         continue
      }
      for j := 0; j < wordSize; j++ {
         if word&(1<<uint(j)) != 0 {
            if buf.Len() > len("{") {
               buf.WriteByte(' ')
            }
            fmt.Fprintf(&buf, "%d", wordSize*i+j)
         }
      }
   }
   buf.WriteByte('}')
   return buf.String()
}

// Return set elements.
func (s *IntSet) Elems() []int {
   e := make([]int, 0)
   for i, word := range s.words {
      for j := 0; j < wordSize; j++ {
         if word&(1<<uint(j)) != 0 {
            e = append(e, i*wordSize+j)
         }
      }
   }
   return e
}

以上所述就是小编给大家介绍的《golang练手小项目系列(1)-位向量》,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对 码农网 的支持!

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

Effective Modern C++ 简体中文版

Effective Modern C++ 简体中文版

Scott Meyers / 高博 / 中国电力出版社 / 2018-4-23 / 99

想要彻底理解C++11和C++14,不可止步于熟悉它们引入的语言特性(例如,auto型别推导、移动语义、lambda表达式以及并发支持)。挑战在于高效地运用这些特性——从而使你的软件具备正确性、高效率、可维护性和可移植性。这正是这本实用的图书意欲达成的定位。它描述的正是使用C++11和C++14——现代C++来撰写真正卓越的软件之道。 涵盖以下主题: 大括号初始化、noexcept规格......一起来看看 《Effective Modern C++ 简体中文版》 这本书的介绍吧!

HTML 编码/解码
HTML 编码/解码

HTML 编码/解码

RGB HSV 转换
RGB HSV 转换

RGB HSV 互转工具

HEX HSV 转换工具
HEX HSV 转换工具

HEX HSV 互换工具