Лента заказов — односвязный список кодов. Функция remove_all(head, code) удаляет все узлы с кодом code (отменённые заказы) и возвращает новую голову. В функции две ошибки — найдите и исправьте их.
# 7 → 0 → 3 → 0 → 0 → 5
remove_all(head, 0) # 7 → 3 → 5
# 0 → 0 → 1
remove_all(head, 0) # 1Заготовка задачи обновилась с тех пор, как вы сохранили свой код.
Сбросить редактор к заготовке? Ваш код будет удалён.
class Node:
"""Узел односвязного списка."""
def __init__(self, value: int, next: "Node | None" = None) -> None:
self.value = value
self.next = next
def remove_all(head: Node | None, code: int) -> Node | None:
"""Удаляет все узлы со значением code, возвращает новую голову."""
if head is not None and head.value == code:
head = head.next
cur = head
while cur is not None and cur.next is not None:
if cur.next.value == code:
cur.next = cur.next.next
cur = cur.next
return headВывод обрезан: показаны первые строки.
def _build(values: list[int]) -> Node | None:
head = None
for v in reversed(values):
head = Node(v, head)
return head
def _values(head: Node | None) -> list[int]:
values = []
while head is not None:
values.append(head.value)
assert len(values) <= 10_000, "в списке цикл"
head = head.next
return values
def _check(values: list[int], code: int) -> None:
got = _values(remove_all(_build(values), code))
expected = [v for v in values if v != code]
assert got == expected, f"remove_all({values}, {code}): ожидалось {expected}, получено {got}"
def test_example():
"""Пример: 7 0 3 0 0 5 без нулей"""
_check([7, 0, 3, 0, 0, 5], 0)
def test_head_run():
"""Удаляемые в начале подряд: 0 0 1"""
_check([0, 0, 1], 0)
def test_single_head():
"""Удаляется только голова"""
_check([4, 1, 2], 4)
def test_tail():
"""Удаляется хвост"""
_check([1, 2, 9], 9)
def test_all_removed():
"""Удаляется всё → None"""
got = remove_all(_build([3, 3, 3]), 3)
assert got is None, f"ожидалось None, получено список {_values(got)}"
def test_nothing_to_remove():
"""Нечего удалять — список не меняется"""
_check([1, 2, 3], 8)
def test_empty():
"""Пустой список → None"""
got = remove_all(None, 1)
assert got is None, f"ожидалось None, получено {got!r}"
def test_mixed_runs():
"""Серии удаляемых в разных местах"""
_check([2, 2, 1, 2, 2, 2, 3, 2, 4, 2, 2], 2)Можно решать и в своём редакторе: «Скопировать с тестами» и запустите python3 remove_all.py — тесты те же, результат покажет ✓ / ✗ и итог.
Запустите тесты. Ошибки проявляются, когда удаляемые элементы стоят подряд.
Что будет, если удалить нужно первые два узла? Условие для головы проверяется один раз.
После удаления cur.next следующий кандидат уже на месте cur.next. Нужно ли двигать cur?
class Node:
"""Узел односвязного списка."""
def __init__(self, value: int, next: "Node | None" = None) -> None:
self.value = value
self.next = next
def remove_all(head: Node | None, code: int) -> Node | None:
"""Удаляет все узлы со значением code, возвращает новую голову."""
while head is not None and head.value == code:
head = head.next
cur = head
while cur is not None and cur.next is not None:
if cur.next.value == code:
cur.next = cur.next.next
else:
cur = cur.next
return head