吃饭

set/hashset/hashset.go

@@ -0,0 +1,80 @@
+package hashset
+
+import (
+	"fmt"
+	"strings"
+)
+
+var nullItem = struct{}{}
+
+// HashSet
+type HashSet struct {
+	hm map[interface{}]struct{}
+}
+
+// New
+func New() *HashSet {
+	return &HashSet{hm: make(map[interface{}]struct{})}
+}
+
+// Add
+func (set *HashSet) Add(items ...interface{}) {
+	for _, item := range items {
+		if _, ok := set.hm[item]; !ok {
+			set.hm[item] = nullItem
+		}
+	}
+}
+
+// Remove
+func (set *HashSet) Remove(items ...interface{}) {
+	for _, item := range items {
+		delete(set.hm, item)
+	}
+}
+
+// Values
+func (set *HashSet) Values() []interface{} {
+	values := make([]interface{}, set.Size())
+	count := 0
+	for item := range set.hm {
+		values[count] = item
+		count++
+	}
+	return values
+}
+
+// Contains
+func (set *HashSet) Contains(item interface{}) bool {
+	if _, contains := set.hm[item]; contains {
+		return true
+	}
+	return false
+}
+
+// Empty
+func (set *HashSet) Empty() bool {
+	return set.Size() == 0
+}
+
+// Clear
+func (set *HashSet) Clear() {
+	set.hm = make(map[interface{}]struct{})
+}
+
+// Size
+func (set *HashSet) Size() int {
+	return len(set.hm)
+}
+
+// String
+func (set *HashSet) String() string {
+	content := "["
+	items := []string{}
+	for k := range set.hm {
+		items = append(items, fmt.Sprintf("%v", k))
+	}
+	content += strings.Join(items, ",")
+	content += "]"
+	return content
+}

set/hashset/hashset_test.go

@@ -0,0 +1,54 @@
+package hashset
+
+import (
+	"bytes"
+	"encoding/gob"
+	"io/ioutil"
+	"log"
+	"testing"
+)
+
+func loadTestData() []int {
+	log.SetFlags(log.Lshortfile)
+
+	data, err := ioutil.ReadFile("../l.log")
+	if err != nil {
+		log.Println(err)
+	}
+	var l []int
+	decoder := gob.NewDecoder(bytes.NewReader(data))
+	decoder.Decode(&l)
+	return l
+}
+
+func TestAdd(t *testing.T) {
+
+	set := New()
+	for i := 0; i < 10; i++ {
+		set.Add(i)
+	}
+
+	if set.Size() != 10 {
+		t.Error("size is not equals to 10")
+	}
+}
+
+func TestRemove(t *testing.T) {
+	set := New()
+
+	for i := 0; i < 10; i++ {
+		set.Add(i)
+	}
+
+	for i := 0; i < 9; i++ {
+		set.Remove(i)
+	}
+
+	if set.Size() != 1 {
+		t.Error("size is not equals to 0")
+	}
+
+	if set.Values()[0] != 9 {
+		t.Error("remain is not 9")
+	}
+}