Lesson 7: Stacks and Queues → Brackets

문제

https://app.codility.com/programmers/lessons/7-stacks_and_queues/brackets/

Brackets coding task - Learn to Code - Codility

 

---

문제 풀이

괄호가 올바른 쌍을 이루는지 확인하는 문제다.

예를 들어 ({)} 이면 올바른 괄호 쌍 이지만 (}){ 일 경우 올바른 괄호 쌍이 아니다.

 

---

코드

def solution(S):
    stack = []
    opens = ('{', '[', '(')
    close = ('}', ']', ')')

    for char in S:
        if char in opens:
            stack.append(char)
        else:
            if not stack:
                return 0
            elif opens.index(stack.pop()) == close.index(char):
                continue
            else:
                return 0

    return 1 if not stack else 0

• 테스트 케이스 중 { { { {가 입력되는 경우가 있다. 이를 위해 stack에 값 존재 여부를 확인하여야 한다.