CSES - AVL-puu

Toteuta luokka TreeSet niin, että sen taustalla on kurssimateriaalin mukaisesti toteutettu AVL-puu.

Toteuta tiedostoon avltree.py luokka TreeSet seuraavan mallin mukaisesti.

class TreeSet:
    def add(self, x):
        # TODO

    def __contains__(self, x):
        # TODO

    def __repr__(self):
        # TODO

if __name__ == "__main__":
    s = TreeSet()
    s.add(1)
    s.add(2)
    s.add(4)
    print(1 in s) # True
    print(2 in s) # True
    print(3 in s) # False
    print(4 in s) # True
    print(s) # [1, 2, 4]