From 24c2952ab16680383864625c8cc8cd73bfd94849 Mon Sep 17 00:00:00 2001 From: Paul Buetow Date: Thu, 5 Nov 2020 09:42:08 +0000 Subject: BST basic unit test works --- set/tree.go | 137 ------------------------------------------------------------ 1 file changed, 137 deletions(-) delete mode 100644 set/tree.go (limited to 'set/tree.go') diff --git a/set/tree.go b/set/tree.go deleted file mode 100644 index 8819791..0000000 --- a/set/tree.go +++ /dev/null @@ -1,137 +0,0 @@ -package set - -type node struct { - key int - val int - left *node - right *node -} - -type Tree struct { - root *node -} - -func NewTree() *Tree { - return &Tree{} -} - -func (t *Tree) Empty() bool { - return t.root == nil -} - -func (t *Tree) Set(key, val int) { - if t.Empty() { - t.root = &node{key, val, nil, nil} - return - } - ptr, _, err := t.search(&t.root, key) - if err != nil { - // key already in the tree as node found - return - } - - *ptr = &node{key, val, nil, nil} -} - -func (t *Tree) Get(key int) (int, error) { - _, n, err := t.search(&t.root, key) - return n.val, err -} - -func (t *Tree) Del(key int) (int, error) { - ptr, n, err := t.search(&t.root, key) - if err != nil { - return 0, err - } - - // Case 1: n is leaf node - // Case 2: n has one child - // Case 3: n has two childs - - switch { - case n.left == nil: - if n.right == nil { - // I am a leaf node - *ptr = nil - return n.val, nil - } - // I have a right child - *ptr = n.right - return n.val, nil - - case n.right == nil: - // I have a left child - *ptr = n.left - return n.val, nil - default: - // I have two children! - - o, err := t.deleteMin(&n.right) - if err != nil { - return 0, err - } - - o.left = n.left - o.right = n.right - *ptr = o - - return n.val, nil - } -} - -func (t *Tree) search(ptr **node, key int) (**node, *node, error) { - n := *ptr - if n == nil { - return nil, nil, NotFound - } - - switch { - case key < n.key: - return t.search(&n.left, key) - case n.key < key: - return t.search(&n.right, key) - default: - return ptr, n, nil - } -} - -func (t *Tree) deleteMin(ptr **node) (*node, error) { - ptr, n, err := t.min(ptr) - if err != nil { - return nil, err - } - - *ptr = n.right - n.right = nil - return n, nil -} - -func (t *Tree) min(ptr **node) (**node, *node, error) { - n := *ptr - if n == nil { - return nil, nil, NotFound - } - - for { - if n.left == nil { - return ptr, n, nil - } - ptr = &n.left - n = *ptr - } -} - -func (t *Tree) max(ptr **node) (**node, *node, error) { - n := *ptr - if n == nil { - return nil, nil, NotFound - } - - for { - if n.right == nil { - return ptr, n, nil - } - ptr = &n.right - n = *ptr - } -} -- cgit v1.2.3