diff --git a/go.sum b/go.sum index b3f692e..fc26243 100644 --- a/go.sum +++ b/go.sum @@ -4,3 +4,4 @@ github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/emirpasic/gods v1.12.0 h1:QAUIPSaCu4G+POclxeqb3F+WPpdKqFGlw36+yOzGlrg= github.com/emirpasic/gods v1.12.0/go.mod h1:YfzfFFoVP/catgzJb4IKIqXjX78Ha8FMSDh3ymbK86o= +github.com/petar/GoLLRB v0.0.0-20190514000832-33fb24c13b99 h1:KcEvVBAvyHkUdFAygKAzwB6LAcZ6LS32WHmRD2VyXMI= diff --git a/tree/tried/tried.go b/tree/tried/tried.go index b4678f2..77f5a6c 100644 --- a/tree/tried/tried.go +++ b/tree/tried/tried.go @@ -1,9 +1,30 @@ package tried +type TriedString string + +func (ts TriedString) Size() uint { + return uint(len(ts)) +} + +func (ts TriedString) WordIndex(idx uint) uint { + w := ts[idx] + if w >= 'a' && w <= 'z' { + return uint(w) - 'a' + } else if w >= 'A' && w <= 'Z' { + return uint(w) - 'A' + 26 + } else { + return uint(w) - '0' + 52 + } +} + +type ObjectIndex interface { + WordIndex(idx uint) uint + Size() uint +} + type Tried struct { root *Node datasize uint - // wordIndex func () } type Node struct { @@ -14,17 +35,28 @@ type Node struct { func New() *Tried { tried := &Tried{} tried.root = new(Node) + tried.datasize = 62 return tried } -func (tried *Tried) Put(words string, values ...interface{}) { +func (tried *Tried) wordIndex(w byte) uint { + if w >= 'a' && w <= 'z' { + return uint(w) - 'a' + } else if w >= 'A' && w <= 'Z' { + return uint(w) - 'A' + 26 + } else { + return uint(w) - '0' + 52 + } +} + +func (tried *Tried) Put(words ObjectIndex, values ...interface{}) { cur := tried.root var n *Node - for i := 0; i < len(words); i++ { - w := uint(words[i] - 'a') + for i := uint(0); i < words.Size(); i++ { + w := words.WordIndex(i) if cur.data == nil { - cur.data = make([]*Node, 26) + cur.data = make([]*Node, tried.datasize) } if n = cur.data[w]; n == nil { @@ -48,11 +80,11 @@ func (tried *Tried) Put(words string, values ...interface{}) { } -func (tried *Tried) Get(words string) interface{} { +func (tried *Tried) Get(words ObjectIndex) interface{} { cur := tried.root var n *Node - for i := 0; i < len(words); i++ { - w := uint(words[i] - 'a') //TODO: 升级Index 函数 + for i := uint(0); i < words.Size(); i++ { + w := words.WordIndex(i) //TODO: 升级Index 函数 if n = cur.data[w]; n == nil { return nil } @@ -61,7 +93,7 @@ func (tried *Tried) Get(words string) interface{} { return n.value } -func (tried *Tried) Has(words string) bool { +func (tried *Tried) Has(words ObjectIndex) bool { return tried.Get(words) != nil } diff --git a/tree/tried/tried_test.go b/tree/tried/tried_test.go index fbc6d1d..9ff23c3 100644 --- a/tree/tried/tried_test.go +++ b/tree/tried/tried_test.go @@ -2,37 +2,40 @@ package tried import ( "testing" + + "github.com/Pallinder/go-randomdata" ) func TestTried_PutAndGet1(t *testing.T) { tried := New() - tried.Put("asdf") - tried.Put("hehe", "hehe") - tried.Put("xixi", 3) + + tried.Put(TriedString("asdf")) + tried.Put(TriedString("hehe"), "hehe") + tried.Put(TriedString("xixi"), 3) var result interface{} - result = tried.Get("asdf") + result = tried.Get(TriedString("asdf")) if result != tried { t.Error("result should be 3") } - result = tried.Get("xixi") + result = tried.Get(TriedString("xixi")) if result != 3 { t.Error("result should be 3") } - result = tried.Get("hehe") + result = tried.Get(TriedString("hehe")) if result != "hehe" { t.Error("result should be hehe") } - result = tried.Get("haha") + result = tried.Get(TriedString("haha")) if result != nil { t.Error("result should be nil") } - result = tried.Get("b") + result = tried.Get(TriedString("b")) if result != nil { t.Error("result should be nil") } @@ -40,10 +43,10 @@ func TestTried_PutAndGet1(t *testing.T) { func TestTried_Traversal(t *testing.T) { tried := New() - tried.Put("asdf") - tried.Put("abdf", "ab") - tried.Put("hehe", "hehe") - tried.Put("xixi", 3) + tried.Put(TriedString("asdf")) + tried.Put(TriedString("abdf"), "ab") + tried.Put(TriedString("hehe"), "hehe") + tried.Put(TriedString("xixi"), 3) var result []interface{} tried.Traversal(func(idx uint, v interface{}) bool { @@ -67,5 +70,49 @@ func TestTried_Traversal(t *testing.T) { if result[3] != 3 { t.Error(result[3]) } - +} + +func BenchmarkTried_Put(b *testing.B) { + + var data []TriedString + b.N = 10000 + count := 1000 + + for i := 0; i < b.N; i++ { + data = append(data, TriedString(randomdata.RandStringRunes(10)+randomdata.RandStringRunes(4))) + } + + b.ResetTimer() + b.N = b.N * count + for c := 0; c < count; c++ { + tried := New() + for _, v := range data { + tried.Put(v) + } + } +} + +func BenchmarkTried_Get(b *testing.B) { + + var data []TriedString + b.N = 10000 + count := 1000 + + for i := 0; i < b.N; i++ { + data = append(data, TriedString(randomdata.RandStringRunes(10)+randomdata.RandStringRunes(4))) + } + + b.N = b.N * count + + tried := New() + for _, v := range data { + tried.Put(v) + } + + b.ResetTimer() + for c := 0; c < count; c++ { + for _, v := range data { + tried.Get(v) + } + } }