217 lines
5.4 KiB
Python
217 lines
5.4 KiB
Python
# phonebook.py
|
|
|
|
# Узел списка: {'n': имя, 'p': телефон, 'nxt': следующий}
|
|
def ll_insert(head, name, phone):
|
|
# обновление, если уже есть
|
|
curr = head
|
|
while curr is not None:
|
|
if curr['n'] == name:
|
|
curr['p'] = phone
|
|
return head
|
|
curr = curr['nxt']
|
|
# вставка в начало (новый узел становится головой)
|
|
new_node = {'n': name, 'p': phone, 'nxt': head}
|
|
return new_node
|
|
|
|
def ll_find(head, name):
|
|
curr = head
|
|
while curr is not None:
|
|
if curr['n'] == name:
|
|
return curr['p']
|
|
curr = curr['nxt']
|
|
return None
|
|
|
|
def ll_delete(head, name):
|
|
if head is None:
|
|
return None
|
|
if head['n'] == name:
|
|
return head['nxt']
|
|
prev = head
|
|
curr = head['nxt']
|
|
while curr is not None:
|
|
if curr['n'] == name:
|
|
prev['nxt'] = curr['nxt']
|
|
return head
|
|
prev = curr
|
|
curr = curr['nxt']
|
|
return head
|
|
|
|
def ll_list_all(head):
|
|
records = []
|
|
curr = head
|
|
while curr is not None:
|
|
records.append((curr['n'], curr['p']))
|
|
curr = curr['nxt']
|
|
records.sort(key=lambda x: x[0])
|
|
return records
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# хеш-функция: сумма ord(name) % size
|
|
def _hash(name, size):
|
|
h = 0
|
|
for ch in name:
|
|
h += ord(ch)
|
|
return h % size
|
|
|
|
SIZE = 13 # фиксированный размер таблицы
|
|
|
|
def ht_create():
|
|
return [None] * SIZE
|
|
|
|
def ht_insert(buckets, name, phone):
|
|
idx = _hash(name, len(buckets))
|
|
buckets[idx] = ll_insert(buckets[idx], name, phone)
|
|
return buckets
|
|
|
|
def ht_find(buckets, name):
|
|
idx = _hash(name, len(buckets))
|
|
return ll_find(buckets[idx], name)
|
|
|
|
def ht_delete(buckets, name):
|
|
idx = _hash(name, len(buckets))
|
|
buckets[idx] = ll_delete(buckets[idx], name)
|
|
return buckets
|
|
|
|
def ht_list_all(buckets):
|
|
all_records = []
|
|
for head in buckets:
|
|
curr = head
|
|
while curr:
|
|
all_records.append((curr['n'], curr['p']))
|
|
curr = curr['nxt']
|
|
all_records.sort(key=lambda x: x[0])
|
|
return all_records
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Узел дерева: {'n': имя, 'p': телефон, 'l': левый, 'r': правый}
|
|
def bst_create_node(name, phone):
|
|
return {'n': name, 'p': phone, 'l': None, 'r': None}
|
|
|
|
def bst_insert(root, name, phone):
|
|
if root is None:
|
|
return bst_create_node(name, phone)
|
|
# итеративная вставка (без рекурсии)
|
|
parent = None
|
|
cur = root
|
|
while cur:
|
|
parent = cur
|
|
if name == cur['n']:
|
|
cur['p'] = phone
|
|
return root
|
|
elif name < cur['n']:
|
|
cur = cur['l']
|
|
else:
|
|
cur = cur['r']
|
|
# вставляем как лист
|
|
if name < parent['n']:
|
|
parent['l'] = bst_create_node(name, phone)
|
|
else:
|
|
parent['r'] = bst_create_node(name, phone)
|
|
return root
|
|
|
|
def bst_find(root, name):
|
|
cur = root
|
|
while cur:
|
|
if name == cur['n']:
|
|
return cur['p']
|
|
elif name < cur['n']:
|
|
cur = cur['l']
|
|
else:
|
|
cur = cur['r']
|
|
return None
|
|
|
|
def _bst_min(node):
|
|
while node['l']:
|
|
node = node['l']
|
|
return node
|
|
|
|
def bst_delete(root, name):
|
|
if root is None:
|
|
return None
|
|
# поиск узла и родителя
|
|
parent = None
|
|
cur = root
|
|
while cur and cur['n'] != name:
|
|
parent = cur
|
|
if name < cur['n']:
|
|
cur = cur['l']
|
|
else:
|
|
cur = cur['r']
|
|
if cur is None:
|
|
return root
|
|
# случай 0 или 1 ребёнок
|
|
if cur['l'] is None or cur['r'] is None:
|
|
child = cur['l'] if cur['l'] else cur['r']
|
|
if parent is None:
|
|
return child
|
|
if parent['l'] == cur:
|
|
parent['l'] = child
|
|
else:
|
|
parent['r'] = child
|
|
else:
|
|
# два ребёнка - ищем inorder-преемника
|
|
succ_parent = cur
|
|
succ = cur['r']
|
|
while succ['l']:
|
|
succ_parent = succ
|
|
succ = succ['l']
|
|
cur['n'], cur['p'] = succ['n'], succ['p']
|
|
if succ_parent['l'] == succ:
|
|
succ_parent['l'] = succ['r']
|
|
else:
|
|
succ_parent['r'] = succ['r']
|
|
return root
|
|
|
|
def bst_list_all(root):
|
|
result = []
|
|
def inorder(node):
|
|
if node:
|
|
inorder(node['l'])
|
|
result.append((node['n'], node['p']))
|
|
inorder(node['r'])
|
|
inorder(root)
|
|
return result
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# TESTING
|
|
|
|
if __name__ == '__main__':
|
|
print("=== Linked list test ===")
|
|
head = None
|
|
head = ll_insert(head, "Ivan", "111")
|
|
head = ll_insert(head, "Anna", "222")
|
|
head = ll_insert(head, "Ivan", "333")
|
|
print(ll_find(head, "Ivan")) # 333
|
|
print(ll_list_all(head)) # [('Anna','222'),('Ivan','333')]
|
|
head = ll_delete(head, "Anna")
|
|
print(ll_list_all(head)) # [('Ivan','333')]
|
|
|
|
print("\n=== Hash table test ===")
|
|
buckets = ht_create()
|
|
ht_insert(buckets, "Ivan", "111")
|
|
ht_insert(buckets, "Boris", "444")
|
|
print(ht_find(buckets, "Ivan")) # 111
|
|
print(ht_list_all(buckets)) # [('Boris','444'),('Ivan','111')]
|
|
|
|
print("\n=== BST test ===")
|
|
root = None
|
|
root = bst_insert(root, "Ivan", "111")
|
|
root = bst_insert(root, "Anna", "222")
|
|
root = bst_insert(root, "Ivan", "333")
|
|
print(bst_find(root, "Ivan")) # 333
|
|
print(bst_list_all(root)) # [('Anna','222'),('Ivan','333')] |