From f0828a8336a4404b8a9807c4aaf6313f4ff8c5e8 Mon Sep 17 00:00:00 2001 From: Paul Buetow Date: Fri, 16 Oct 2020 09:35:52 +0100 Subject: more on tree --- set/tree.go | 17 +++++++++++++---- 1 file 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 +} -- cgit v1.2.3