Extend the class TreeSet
by adding the methods prev
and next
that return the predecessor and the successor elements as described in the course material. If there is no such element, return None
.
In a file prevnext.py
, implement the class TreeSet
according to the following template.
class TreeSet: # methods in the course material def prev(self, x): # TODO def next(self, x): # TODO if __name__ == "__main__": s = TreeSet() s.add(2) s.add(5) print(s.prev(5)) # 2 print(s.prev(2)) # None print(s.next(1)) # 2 print(s.next(2)) # 5 print(s.next(5)) # None