Python: Поиск в двоичном дереве

src/solution.py

Двоичное дерево поиска состоит из узлов, каждый из которых содержит значение ключа и два поддерева (левое и правое), которые в свою очередь также являются двоичными деревьями. Правильное дерево не содержит повторяющихся ключей, и для каждого узла гарантируется, что в левом поддереве все значения меньше текущего, а в правом — больше.

Реализуйте класс, который реализует представление узла. При инициализации объекта класс принимает на вход три параметра:

• key — значение ключа (число),
• left — левое поддерево (тоже узел, по умолчанию None),
• right — правое поддерево (по умолчанию None).

Каждый экземпляр класса должен содержать атрибуты:

• key
• left
• right

Также класс должен реализовывать метод search(key), который выполняет поиск узла в правильно построенном двоичном дереве по ключу и возвращает узел. Если узел не найден, возвращается None.

>>> from solution import Node
>>> node5 = Node(5)
>>> node22 = Node(22, right=Node(20))
>>> tree = Node(
...     9,
...     Node(
...         4,
...         Node(3),
...         Node(
...             6,
...             node5,
...             Node(7),
...         ),
...     ),
...     Node(
...         17,
...         right=node22,
...     ),
... )
>>> tree.search(6).key
6
>>> tree.search(6).left.key
5
>>> tree.search(6).right.key
7
>>> tree.search(5) is node5
True
>>> tree.left.left.key
3
>>>

Подсказки

• Двоичное дерево поиска