78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
|
package treemap
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/davecgh/go-spew/spew"
|
||
|
|
|
||
|
|
"github.com/474420502/focus/compare"
|
||
|
|
"github.com/474420502/focus/tree/avlkeydup"
|
||
|
|
)
|
||
|
|
|
||
|
|
type TreeMap struct {
|
||
|
|
avl *avlkeydup.Tree
|
||
|
|
}
|
||
|
|
|
||
|
|
// New instantiates a hash map.
|
||
|
|
func New(Compare compare.Compare) *TreeMap {
|
||
|
|
return &TreeMap{avl: avlkeydup.New(Compare)}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Put inserts element into the map.
|
||
|
|
func (tmap *TreeMap) Put(key interface{}, value interface{}) {
|
||
|
|
tmap.avl.Put(key, value)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (tmap *TreeMap) Get(key interface{}) (value interface{}, isfound bool) {
|
||
|
|
value, isfound = tmap.avl.Get(key)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (tmap *TreeMap) Remove(key interface{}) {
|
||
|
|
tmap.Remove(key)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (tmap *TreeMap) Empty() bool {
|
||
|
|
return tmap.avl.Size() == 0
|
||
|
|
}
|
||
|
|
|
||
|
|
func (tmap *TreeMap) Size() int {
|
||
|
|
return tmap.avl.Size()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (tmap *TreeMap) Keys() []interface{} {
|
||
|
|
keys := make([]interface{}, tmap.avl.Size())
|
||
|
|
count := 0
|
||
|
|
tmap.avl.Traversal(func(key, value interface{}) bool {
|
||
|
|
keys[count] = key
|
||
|
|
count++
|
||
|
|
return true
|
||
|
|
})
|
||
|
|
return keys
|
||
|
|
}
|
||
|
|
|
||
|
|
func (tmap *TreeMap) Values() []interface{} {
|
||
|
|
values := make([]interface{}, tmap.avl.Size())
|
||
|
|
count := 0
|
||
|
|
tmap.avl.Traversal(func(key, value interface{}) bool {
|
||
|
|
values[count] = value
|
||
|
|
count++
|
||
|
|
return true
|
||
|
|
})
|
||
|
|
return values
|
||
|
|
}
|
||
|
|
|
||
|
|
func (tmap *TreeMap) Clear() {
|
||
|
|
tmap.avl.Clear()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (tmap *TreeMap) String() string {
|
||
|
|
content := "{"
|
||
|
|
tmap.avl.Traversal(func(key, value interface{}) bool {
|
||
|
|
content += spew.Sprint(key) + ":" + spew.Sprint(value) + ","
|
||
|
|
return true
|
||
|
|
})
|
||
|
|
content = strings.TrimRight(content, ",") + "}"
|
||
|
|
return content
|
||
|
|
}
|