diff options
| -rw-r--r-- | set/tree.go | 17 |
1 files changed, 13 insertions, 4 deletions
diff --git a/set/tree.go b/set/tree.go index d4f7b7d..8233b0e 100644 --- a/set/tree.go +++ b/set/tree.go @@ -66,7 +66,16 @@ func (t *Tree) Get(key int) (int, error) { return t.get(t.root, key) } -/* - Get(key int) (int, error) - Del(key int) (int, error) -*/ +func (t *Tree) Del(key int) (int, error) { + if t.root == nil { + return 0, NotFound + } + + if t.root.key == key { + val := t.root.val + + return val, nil + } + + return 0, NotImplemented +} |