Полностью реализован BinaryTree
This commit is contained in:
parent
b383ed1cdc
commit
dd4eca8407
|
|
@ -14,8 +14,9 @@ def bst_insert(root: dict|None, name: str, phone: str) -> dict:
|
||||||
current = root
|
current = root
|
||||||
while True:
|
while True:
|
||||||
if current['name'] == name:
|
if current['name'] == name:
|
||||||
current['phone'] == phone
|
current['phone'] = phone
|
||||||
elif current['name'] < name:
|
return root
|
||||||
|
elif name < current['name']:
|
||||||
if current['left'] == None:
|
if current['left'] == None:
|
||||||
current['left'] = bst_insert(None, name, phone)
|
current['left'] = bst_insert(None, name, phone)
|
||||||
return root
|
return root
|
||||||
|
|
@ -32,10 +33,56 @@ def bst_insert(root: dict|None, name: str, phone: str) -> dict:
|
||||||
|
|
||||||
def bst_find(root: dict|None, name: str) -> str|None:
|
def bst_find(root: dict|None, name: str) -> str|None:
|
||||||
"""Поиск в ширину."""
|
"""Поиск в ширину."""
|
||||||
|
node = find_node_to_delete(root, name)
|
||||||
|
if node != None:
|
||||||
|
return node['phone']
|
||||||
|
|
||||||
|
def find_node_to_delete(root: dict, name: str) -> dict|None:
|
||||||
|
"""Поиск в ширину."""
|
||||||
|
while root != None:
|
||||||
|
if root['name'] == name:
|
||||||
|
return root
|
||||||
|
elif name < root['name']:
|
||||||
|
root = root['left']
|
||||||
|
else:
|
||||||
|
root = root['right']
|
||||||
|
return None
|
||||||
|
|
||||||
|
def find_minimal_child(root: dict) -> dict|None:
|
||||||
|
while root['left']:
|
||||||
|
root = root['left']
|
||||||
|
return root
|
||||||
|
|
||||||
|
def bst_delete(root: dict, name: str) -> None:
|
||||||
|
"""Удаляет узел и возвращает новый корень."""
|
||||||
|
if root is None:
|
||||||
|
return None
|
||||||
|
|
||||||
|
if name < root['name']:
|
||||||
|
root['left'] = bst_delete(root['left'], name)
|
||||||
|
elif name > root['name']:
|
||||||
|
root['right'] = bst_delete(root['right'], name)
|
||||||
|
else:
|
||||||
|
# Случай 1: нет детей или один ребенок
|
||||||
|
if root['left'] is None:
|
||||||
|
return root['right']
|
||||||
|
elif root['right'] is None:
|
||||||
|
return root['left']
|
||||||
|
|
||||||
|
# Случай 2: два ребенка
|
||||||
|
min_node = find_minimal_child(root['right'])
|
||||||
|
root['name'] = min_node['name']
|
||||||
|
root['phone'] = min_node['phone']
|
||||||
|
root['right'] = bst_delete(root['right'], min_node['name'])
|
||||||
|
|
||||||
|
return root
|
||||||
|
|
||||||
def bst_delete(root: dict, name: str) -> dict:
|
|
||||||
"""Удаление, возвращает новый корень."""
|
|
||||||
|
|
||||||
def bst_list_all(root: dict) -> list:
|
def bst_list_all(root: dict) -> list:
|
||||||
"""Центрированный обход.
|
"""Центрированный обход.
|
||||||
Рекурсивно собирает записи в отсортированном порядке."""
|
Рекурсивно собирает записи в отсортированном порядке."""
|
||||||
|
|
||||||
|
if root is None:
|
||||||
|
return []
|
||||||
|
node_values = {"name": root['name'], "phone": root['phone']}
|
||||||
|
return bst_list_all(root['left']) + [node_values] + bst_list_all(root['right'])
|
||||||
Loading…
Reference in New Issue
Block a user