Кафе на карте — точки с координатами. Функция nearest(points, query) возвращает индекс ближайшей к query точки по евклидову расстоянию. При равных расстояниях — меньший индекс, для пустого списка — -1. В функции две ошибки — найдите и исправьте их.
nearest([[0, 0], [5, 5], [1, 2]], [2, 2]) # 2
nearest([[1, 0], [-1, 0]], [0, 0]) # 0 — ничья, берём меньший индексЗаготовка задачи обновилась с тех пор, как вы сохранили свой код.
Сбросить редактор к заготовке? Ваш код будет удалён.
import math
def nearest(points: list[list[float]], query: list[float]) -> int:
"""Индекс ближайшей точки (меньший при равенстве) или -1."""
best, best_dist = -1, math.inf
for i, point in enumerate(points):
d = math.sqrt(sum(x - y for x, y in zip(point, query)) ** 2)
if d <= best_dist:
best, best_dist = i, d
return bestВывод обрезан: показаны первые строки.
def test_example():
"""Пример: ближайшая — [1, 2]"""
got = nearest([[0, 0], [5, 5], [1, 2]], [2, 2])
assert got == 2, f"ожидалось 2, получено {got!r}"
def test_tie_smaller_index():
"""Ничья → меньший индекс"""
got = nearest([[1, 0], [-1, 0]], [0, 0])
assert got == 0, f"ожидалось 0, получено {got!r}"
def test_differences_do_not_cancel():
"""Разности разного знака не компенсируют друг друга"""
got = nearest([[5, -5], [1, 1]], [0, 0])
assert got == 1, f"ожидалось 1 ([1, 1] ближе, чем [5, -5]), получено {got!r}"
def test_empty():
"""Нет кафе → -1"""
got = nearest([], [0, 0])
assert got == -1, f"ожидалось -1, получено {got!r}"
def test_single():
"""Одно кафе → 0"""
got = nearest([[100, 100]], [0, 0])
assert got == 0, f"ожидалось 0, получено {got!r}"
def test_exact_match():
"""Точка совпадает с запросом"""
got = nearest([[3, 3], [1, 1], [1, 1]], [1, 1])
assert got == 1, f"ожидалось 1, получено {got!r}"
def test_three_dimensions():
"""Три координаты"""
got = nearest([[0, 0, 9], [2, 2, 2], [0, 5, 0]], [1, 1, 1])
assert got == 1, f"ожидалось 1, получено {got!r}"Можно решать и в своём редакторе: «Скопировать с тестами» и запустите python3 nearest_cafe.py — тесты те же, результат покажет ✓ / ✗ и итог.
Запустите тесты. Посмотрите на пример с точкой [5, -5]: какое «расстояние» до неё считает функция?
Сравните формулу в коде с формулой из главы: что возводится в квадрат — каждая разность или их сумма?
При равных расстояниях нужен меньший индекс. Меняется ли лучший индекс при сравнении <=?
import math
def nearest(points: list[list[float]], query: list[float]) -> int:
"""Индекс ближайшей точки (меньший при равенстве) или -1."""
best, best_dist = -1, math.inf
for i, point in enumerate(points):
d = math.sqrt(sum((x - y) ** 2 for x, y in zip(point, query)))
if d < best_dist:
best, best_dist = i, d
return best