forked from UNN/2026-rff_mp
[1] бст
This commit is contained in:
parent
673c18cfaf
commit
7fafaba9eb
|
|
@ -103,6 +103,136 @@ def ht_list_all(buckets: List[Optional[Dict]]) -> List[Tuple[str, str]]:
|
||||||
|
|
||||||
records.sort(key=lambda x: x[0])
|
records.sort(key=lambda x: x[0])
|
||||||
return records
|
return records
|
||||||
|
def bst_insert(root: Optional[Dict], name: str, phone: str) -> Dict:
|
||||||
|
"""Вставка в BST (итеративная)"""
|
||||||
|
new_node = {'name': name, 'phone': phone, 'left': None, 'right': None}
|
||||||
|
|
||||||
|
if root is None:
|
||||||
|
return new_node
|
||||||
|
|
||||||
|
current = root
|
||||||
|
while True:
|
||||||
|
if name < current['name']:
|
||||||
|
if current['left'] is None:
|
||||||
|
current['left'] = new_node
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
current = current['left']
|
||||||
|
elif name > current['name']:
|
||||||
|
if current['right'] is None:
|
||||||
|
current['right'] = new_node
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
current = current['right']
|
||||||
|
else:
|
||||||
|
# Обновляем существующую запись
|
||||||
|
current['phone'] = phone
|
||||||
|
break
|
||||||
|
|
||||||
|
return root
|
||||||
|
|
||||||
|
|
||||||
|
def bst_find(root: Optional[Dict], name: str) -> Optional[str]:
|
||||||
|
"""Поиск в BST (итеративный)"""
|
||||||
|
current = root
|
||||||
|
while current is not None:
|
||||||
|
if name == current['name']:
|
||||||
|
return current['phone']
|
||||||
|
elif name < current['name']:
|
||||||
|
current = current['left']
|
||||||
|
else:
|
||||||
|
current = current['right']
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def bst_min_node(node: Dict) -> Dict:
|
||||||
|
"""Поиск узла с минимальным значением"""
|
||||||
|
current = node
|
||||||
|
while current['left'] is not None:
|
||||||
|
current = current['left']
|
||||||
|
return current
|
||||||
|
|
||||||
|
|
||||||
|
def bst_delete(root: Optional[Dict], name: str) -> Optional[Dict]:
|
||||||
|
"""Удаление из BST (итеративная версия)"""
|
||||||
|
if root is None:
|
||||||
|
return None
|
||||||
|
|
||||||
|
# Поиск узла для удаления и его родителя
|
||||||
|
parent = None
|
||||||
|
current = root
|
||||||
|
|
||||||
|
while current is not None and current['name'] != name:
|
||||||
|
parent = current
|
||||||
|
if name < current['name']:
|
||||||
|
current = current['left']
|
||||||
|
else:
|
||||||
|
current = current['right']
|
||||||
|
|
||||||
|
if current is None:
|
||||||
|
return root
|
||||||
|
|
||||||
|
# Случай 1: узел не имеет детей
|
||||||
|
if current['left'] is None and current['right'] is None:
|
||||||
|
if parent is None:
|
||||||
|
return None
|
||||||
|
if parent['left'] == current:
|
||||||
|
parent['left'] = None
|
||||||
|
else:
|
||||||
|
parent['right'] = None
|
||||||
|
return root
|
||||||
|
|
||||||
|
# Случай 2: узел имеет одного ребёнка
|
||||||
|
if current['left'] is None:
|
||||||
|
child = current['right']
|
||||||
|
elif current['right'] is None:
|
||||||
|
child = current['left']
|
||||||
|
else:
|
||||||
|
# Случай 3: узел имеет двух детей
|
||||||
|
# Находим минимальный узел в правом поддереве
|
||||||
|
successor_parent = current
|
||||||
|
successor = current['right']
|
||||||
|
while successor['left'] is not None:
|
||||||
|
successor_parent = successor
|
||||||
|
successor = successor['left']
|
||||||
|
|
||||||
|
# Копируем данные из successor в current
|
||||||
|
current['name'] = successor['name']
|
||||||
|
current['phone'] = successor['phone']
|
||||||
|
|
||||||
|
# Удаляем successor
|
||||||
|
if successor_parent['left'] == successor:
|
||||||
|
successor_parent['left'] = successor['right']
|
||||||
|
else:
|
||||||
|
successor_parent['right'] = successor['right']
|
||||||
|
|
||||||
|
return root
|
||||||
|
|
||||||
|
# Присоединяем ребёнка к родителю
|
||||||
|
if parent is None:
|
||||||
|
return child
|
||||||
|
if parent['left'] == current:
|
||||||
|
parent['left'] = child
|
||||||
|
else:
|
||||||
|
parent['right'] = child
|
||||||
|
|
||||||
|
return root
|
||||||
|
|
||||||
|
|
||||||
|
def bst_inorder(root: Optional[Dict], records: List[Tuple[str, str]]) -> None:
|
||||||
|
"""Центрированный обход BST (рекурсивный)"""
|
||||||
|
if root is not None:
|
||||||
|
bst_inorder(root['left'], records)
|
||||||
|
records.append((root['name'], root['phone']))
|
||||||
|
bst_inorder(root['right'], records)
|
||||||
|
|
||||||
|
|
||||||
|
def bst_list_all(root: Optional[Dict]) -> List[Tuple[str, str]]:
|
||||||
|
"""Сбор всех записей из BST (уже отсортированы)"""
|
||||||
|
records = []
|
||||||
|
bst_inorder(root, records)
|
||||||
|
return records
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
records.sort(key=lambda x: x[0])
|
records.sort(key=lambda x: x[0])
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user