Writing tests for bracket sequence validation (part 8b4f)

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

Задача

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

Решение

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  # и второй случай

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

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

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

скобки · проверка · ломается · крайние · тестами · работы · Проверить · добавлением · return · правках · базовая · символов · правильно · времени · случаи · защищает · минимум · проверять · Проверка · pairs · памяти · assert · поэтому · делает · конце