eson
368447c31d
new file: LICENSE new file: README.md new file: compare/compare.go new file: for_test.go new file: heap/heap.go new file: heap/heap_test.go new file: interface.go new file: lastack/lastack.go new file: lastack/lastack_test.go new file: list/list.go new file: list/list_test.go new file: map/hashmap/hashmap.go new file: map/hashmap/hashmap_test.go new file: priority_list/iterator.go new file: priority_list/priority_list.go new file: priority_list/priority_list_test.go new file: priority_queue/iterator.go new file: priority_queue/priority_queue.go new file: priority_queue/priority_queue_test.go new file: priority_queue/vbt.go new file: priority_queue/vbt_test.go new file: priority_queuekey/iterator.go new file: priority_queuekey/priority_queuekey.go new file: priority_queuekey/priority_queuekey_test.go new file: priority_queuekey/vbt.go new file: priority_queuekey/vbt_test.go new file: set/hashset/hashset.go new file: set/hashset/hashset_test.go new file: set/treeset/treeset.go new file: set/treeset/treeset_test.go new file: sparse_array/array2/array2.go new file: sparse_array/array2/array2_test.go new file: sparse_array/array3/array3.go new file: sparse_array/array3/array3_test.go new file: sparse_array/arrayn/arrayn.go new file: sparse_array/arrayn/arrayn_test.go new file: stack/stack.go new file: stack/stack_test.go new file: tree/avl/avl.go new file: tree/avl/avl_test.go new file: tree/avl/iterator.go new file: tree/avldup/avldup.go new file: tree/avldup/avldup_test.go new file: tree/avldup/iterator.go new file: tree/avlkey/avlkey.go new file: tree/avlkey/avlkey_test.go new file: tree/avlkey/iterator.go new file: tree/avlkeydup/avlkeydup.go new file: tree/avlkeydup/avlkeydup_test.go new file: tree/avlkeydup/iterator.go new file: tree/vbt/iterator.go new file: tree/vbt/vbt.go new file: tree/vbt/vbt_test.go new file: tree/vbtkey/iterator.go new file: tree/vbtkey/vbtkey.go new file: tree/vbtkey/vbtkey_test.go
86 lines
1.5 KiB
Go
86 lines
1.5 KiB
Go
package treeset
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"474420502.top/eson/structure/avldup"
|
|
"474420502.top/eson/structure/compare"
|
|
)
|
|
|
|
// TreeSet
|
|
type TreeSet struct {
|
|
tree *avldup.Tree
|
|
}
|
|
|
|
// New
|
|
func New(Compare compare.Compare) *TreeSet {
|
|
return &TreeSet{tree: avldup.New(Compare)}
|
|
}
|
|
|
|
// Add
|
|
func (set *TreeSet) Add(items ...interface{}) {
|
|
for _, item := range items {
|
|
set.tree.Put(item)
|
|
}
|
|
}
|
|
|
|
// Remove
|
|
func (set *TreeSet) Remove(items ...interface{}) {
|
|
for _, item := range items {
|
|
set.tree.Remove(item)
|
|
}
|
|
}
|
|
|
|
// Values
|
|
func (set *TreeSet) Values() []interface{} {
|
|
return set.tree.Values()
|
|
}
|
|
|
|
// Contains
|
|
func (set *TreeSet) Contains(item interface{}) bool {
|
|
if _, ok := set.tree.Get(item); ok {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
// Contains the result is [r1,r2], not [r1, r2)
|
|
func (set *TreeSet) GetRange(r1, r2 interface{}) (result []interface{}) {
|
|
return set.tree.GetRange(r1, r2)
|
|
}
|
|
|
|
// Contains the result is [r1,item,r2] r1->item->r2 are close-knit
|
|
func (set *TreeSet) GetAround(item interface{}) (result [3]interface{}) {
|
|
return set.tree.GetAround(item)
|
|
}
|
|
|
|
// Empty
|
|
func (set *TreeSet) Empty() bool {
|
|
return set.Size() == 0
|
|
}
|
|
|
|
// Clear
|
|
func (set *TreeSet) Clear() {
|
|
set.tree.Clear()
|
|
}
|
|
|
|
// Size
|
|
func (set *TreeSet) Size() int {
|
|
return set.tree.Size()
|
|
}
|
|
|
|
// String
|
|
func (set *TreeSet) String() string {
|
|
content := "HashSet\n"
|
|
items := []string{}
|
|
|
|
set.tree.Traversal(func(k interface{}) bool {
|
|
items = append(items, fmt.Sprintf("%v", k))
|
|
return true
|
|
})
|
|
|
|
content += strings.Join(items, ", ")
|
|
return content
|
|
}
|