[1,2] lab1, lab2 #273
144
soldatkinao/lab1/task-1.py
Normal file
144
soldatkinao/lab1/task-1.py
Normal file
|
|
@ -0,0 +1,144 @@
|
|||
# Task 1
|
||||
import time
|
||||
import random
|
||||
import csv
|
||||
from collections import defaultdict
|
||||
|
||||
def ll_insert(head, name, phone):
|
||||
new = {'name': name, 'phone': phone, 'next': None}
|
||||
if head is None:
|
||||
return new
|
||||
cur = head
|
||||
while cur:
|
||||
if cur['name'] == name:
|
||||
cur['phone'] = phone
|
||||
return head
|
||||
if cur['next'] is None:
|
||||
break
|
||||
cur = cur['next']
|
||||
cur['next'] = new
|
||||
return head
|
||||
|
||||
def ll_find(head, name):
|
||||
cur = head
|
||||
while cur:
|
||||
if cur['name'] == name:
|
||||
return cur['phone']
|
||||
cur = cur['next']
|
||||
return None
|
||||
|
||||
def ll_delete(head, name):
|
||||
if head is None:
|
||||
return None
|
||||
if head['name'] == name:
|
||||
return head['next']
|
||||
prev = head
|
||||
cur = head['next']
|
||||
while cur:
|
||||
if cur['name'] == name:
|
||||
prev['next'] = cur['next']
|
||||
return head
|
||||
prev = cur
|
||||
cur = cur['next']
|
||||
return head
|
||||
|
||||
def ll_list_all(head):
|
||||
rec = []
|
||||
cur = head
|
||||
while cur:
|
||||
rec.append((cur['name'], cur['phone']))
|
||||
cur = cur['next']
|
||||
rec.sort(key=lambda x: x[0])
|
||||
return rec
|
||||
|
||||
def gen_data(N=10000):
|
||||
data = []
|
||||
for i in range(N):
|
||||
name = f"user_{i:06d}"
|
||||
phone = f"+7-{random.randint(100,999)}-{random.randint(100,999)}-{random.randint(1000,9999)}"
|
||||
data.append((name, phone))
|
||||
shuffled = data[:]
|
||||
random.shuffle(shuffled)
|
||||
sorted_data = sorted(data, key=lambda x: x[0])
|
||||
return shuffled, sorted_data
|
||||
|
||||
# 5 повторений сохраняем все замеры
|
||||
def run_test(init_func, ins_func, find_func, del_func, list_func,
|
||||
shuffled, sorted_data, exist_names, missing_names, del_names):
|
||||
"""
|
||||
Возвращает список строк для CSV:
|
||||
[порядок, повторение, операция, время]
|
||||
"""
|
||||
rows = []
|
||||
for order, dataset in [('shuffled', shuffled), ('sorted', sorted_data)]:
|
||||
for rep in range(5):
|
||||
s = init_func()
|
||||
# вставка
|
||||
t1 = time.perf_counter()
|
||||
for name, phone in dataset:
|
||||
s = ins_func(s, name, phone)
|
||||
t2 = time.perf_counter()
|
||||
rows.append([order, rep+1, 'insert', t2 - t1])
|
||||
# поиск
|
||||
t1 = time.perf_counter()
|
||||
for name in exist_names:
|
||||
find_func(s, name)
|
||||
for name in missing_names:
|
||||
find_func(s, name)
|
||||
t2 = time.perf_counter()
|
||||
rows.append([order, rep+1, 'find', t2 - t1])
|
||||
# удаление
|
||||
t1 = time.perf_counter()
|
||||
for name in del_names:
|
||||
s = del_func(s, name)
|
||||
t2 = time.perf_counter()
|
||||
rows.append([order, rep+1, 'delete', t2 - t1])
|
||||
|
||||
t1 = time.perf_counter()
|
||||
list_func(s)
|
||||
t2 = time.perf_counter()
|
||||
rows.append([order, rep+1, 'list_all', t2 - t1])
|
||||
return rows
|
||||
|
||||
if __name__ == '__main__':
|
||||
N = 10000
|
||||
print(f"Генерация {N} записей...")
|
||||
shuffled, sorted_data = gen_data(N)
|
||||
|
||||
exist_names = [name for name, _ in shuffled[:100]]
|
||||
missing_names = [f"none_{i}" for i in range(10)]
|
||||
all_names = [name for name, _ in shuffled]
|
||||
del_names = random.sample(all_names, 50)
|
||||
|
||||
all_rows = [] # для CSV
|
||||
|
||||
print("Связный список...")
|
||||
rows = run_test(lambda: None, ll_insert, ll_find, ll_delete, ll_list_all,
|
||||
shuffled, sorted_data, exist_names, missing_names, del_names)
|
||||
for r in rows:
|
||||
all_rows.append(['LinkedList'] + r)
|
||||
|
||||
# Запись в CSV
|
||||
with open('results.csv', 'w', newline='', encoding='utf-8') as f:
|
||||
writer = csv.writer(f)
|
||||
writer.writerow(['Структура', 'Порядок', 'Повторение', 'Операция', 'Время (с)'])
|
||||
writer.writerows(all_rows)
|
||||
|
||||
print("Все замеры сохранены в results.csv")
|
||||
|
||||
# Подсчёт и вывод средних
|
||||
avg = defaultdict(list)
|
||||
for row in all_rows:
|
||||
struct, order, rep, op, t = row
|
||||
avg[(struct, order, op)].append(t)
|
||||
print("\nСредние значения (5 запусков):")
|
||||
print(f"{'Структура':<12} {'Порядок':<10} {'Вставка':>10} {'Поиск':>10} {'Удаление':>10} {'ListAll':>10}")
|
||||
for struct in ['LinkedList']:
|
||||
for order in ['shuffled', 'sorted']:
|
||||
row = [struct, order]
|
||||
for op in ['insert', 'find', 'delete', 'list_all']:
|
||||
vals = avg[(struct, order, op)]
|
||||
avg_val = sum(vals)/len(vals)
|
||||
row.append(f"{avg_val:.5f}")
|
||||
print(f"{row[0]:<12} {row[1]:<10} {row[2]:>10} {row[3]:>10} {row[4]:>10} {row[5]:>10}")
|
||||
print("\nГотово (только связный список).")
|
||||
Loading…
Reference in New Issue
Block a user