Writing tests for bracket sequence validation (part 191c)

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

Задача

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

Решение

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

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

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

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

Хорошая · Проверить · вершиной · pairs · вызова · видно · stack.append(ch · Проверка · Разберём · хороший · сразу · stack · времени · Решение · случаи · символов · худшем · [pairsch · строке · return · случайной · накроем · проверять · assert · стека · нескольких · открывающая · закрывающая · случай