Writing tests for bracket sequence validation (part 55a7)

Разберём решение и сразу накроем его тестами — это хороший стиль работы с кодом.

Задача

Проверить, правильно ли расставлены скобки в строке из символов ()[]{}.

Решение

def is_balanced(s: str) -> bool:
    pairs = {')': '(', ']': '[', '}': '{'}
    stack = []
    for ch in s:
        if ch in '([{':
            stack.append(ch)
        elif ch in pairs:
            if not stack or stack.pop() != pairs[ch]:
                return False
    return not stack


print(is_balanced("([]{})") )

Как это работает

Это классическое применение стека: открывающая скобка кладётся наверх, закрывающая сверяется с вершиной. Если в конце стек пуст — последовательность верна.

Пример работы

True

Сложность

O(n) по времени, O(n) по памяти в худшем случае.

Проверка тестами

Хорошая привычка — проверять решение сразу на нескольких примерах, включая крайние случаи.

assert is_valid(True) is not None  # базовая проверка вызова
assert is_valid(False) is not None  # и второй случай

Такой минимум уже защищает от случайной ошибки при правках кода.

С добавлением проверок решение становится самодокументируемым — сразу видно, что код делает.

Читайте также:

Задача · классическое · [pairsch · закрывающая · верна · памяти · минимум · делает · Сложность · False · худшем · стиль · кодом · stack · assert · конце · случаи · защищает · кладётся · символов · привычка · становится · хороший · правках · python