[코테] 스택 - 괄호 회전하기

문제:

출처: 프로그래머스 - 괄호 회전하기

문제 설명
다음 규칙을 지키는 문자열을 올바른 괄호 문자열이라고 정의합니다.

(), [], {} 는 모두 올바른 괄호 문자열입니다.
만약 A가 올바른 괄호 문자열이라면, (A), [A], {A} 도 올바른 괄호 문자열입니다. 예를 들어, [] 가 올바른 괄호 문자열이므로, ([]) 도 올바른 괄호 문자열입니다.
만약 A, B가 올바른 괄호 문자열이라면, AB 도 올바른 괄호 문자열입니다. 예를 들어, {} 와 ([]) 가 올바른 괄호 문자열이므로, {}([]) 도 올바른 괄호 문자열입니다.
대괄호, 중괄호, 그리고 소괄호로 이루어진 문자열 s가 매개변수로 주어집니다. 이 s를 왼쪽으로 x (0 ≤ x < (s의 길이)) 칸만큼 회전시켰을 때 s가 올바른 괄호 문자열이 되게 하는 x의 개수를 return 하도록 solution 함수를 완성해주세요.

제한사항
s의 길이는 1 이상 1,000 이하입니다.

입출력 예 #

다음 표는 "[](){}" 를 회전시킨 모습을 나타낸 것입니다.
x	s를 왼쪽으로 x칸만큼 회전	올바른 괄호 문자열?
0	"[](){}"	O
1	"](){}["	X
2	"(){}[]"	O
3	"){}[]("	X
4	"{}[]()"	O
5	"}[](){"	X
올바른 괄호 문자열이 되는 x가 3개이므로, 3을 return 해야 합니다.

풀이:

1. 올바른 괄호인지 아닌지 판별하는 함수 작성.
   1.1. '( { [' 와 같이 여는 괄호가 나올 시 stack에 추가
   1.2. ') } ]' 와 같이 닫는 괄호가 나올 시 stack[-1](스택에 가장 마지막)의 여는 괄호와 짝이 맞다면 stack에서 빼낸다. 짝이 맞지 않다면 올바른 괄호가 아니므로 False로 함수 종료.
   1.3. 모든 괄호를 다 살펴본 후 stack가 비었다면 짝이 맞았으므로 True반환, stack에 괄호가 남아있다면 짝이 맞지 않았으므로 False 반환.
2. 올바른 괄호인지 판변 후, 올바른 괄호라면 횟수를 추가.

코드:

def getTF(s): # 올바른 함수인지 판별하여 True/False 반환
    stack = []
    for i in s:
        if i in ('(', '{', '['): # 여는 괄호가 나왔을 때
            stack.append(i)
        else: # 닫는 괄호가 나왔을 때
            if len(stack) == 0: # 스택이 비어있으면 짝이 맞지 않음으로 False
                return False
            else: # 스택에 있는 괄호와 짝이 맞다면 빼낸다.
                if i == ')' and stack[-1] == '(':
                    stack.pop()
                elif i == '}' and stack[-1] == '{':
                    stack.pop()
                elif i == ']' and stack[-1] == '[':
                    stack.pop()
                else: # 스택에 있는 괄호와 짝이 맞지 않으면 잘못된 괄호이므로 Flase 반환.
                    return False
    # 모든 괄호를 확인한 후, 
    if len(stack) == 0: #짝이 맞은 경우
        return True
    else: # 짝이 틀린 경우
        return False

def solution(s):
    n = len(s)
    s = list(s)
    answer = 0
    for i in range(n):
        s.append(s.pop(0))
        if getTF(s): # 올바른 괄호라면
            answer += 1
            
    return answer