[Programmers] [월간 코드 챌린지 시즌2] [Python] Level2_괄호 회전하기

문제

https://school.programmers.co.kr/learn/courses/30/lessons/76502

프로그래머스

풀이

# 프로그래머스 2단계 - 월간 코드 챌린지 시즌2 - 괄호 회전하기
from collections import deque
def check(s):
    answer = 0
    stack = []
    for i in s:
        if len(stack) == 0:
            stack.append(i)
        else:
            if i == ')' and stack[-1] == '(':
                stack.pop()
            elif i == '}' and stack[-1] == '{':
                stack.pop()
            elif i == ']' and stack[-1] == '[':
                stack.pop()
            else:
                stack.append(i)
    return 1 if len(stack) == 0 else 0
def solution(s):
    answer = 0
    for i in range(len(s)):
        if check(s): answer += 1
        s = s[1:] + s[:1]
    return answer

solution("[](){}")

비슷한 문제로 올바른 괄호가 있다. 올바른 괄호를 확인하기 위해 스택을 사용하였는데 여기서도 또한 스택을 사용하였다. 

열린 괄호가 나오면 스택의 끝에 추가 해주고 닫힌 괄호가 나오면 스택에 있는 원소를 확인해 같은 괄호면 스택에서 제거해 준다.