스택 - 괄호 회전하기

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

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

“( )”, “[ ]”, “{ }”는 모두 올바른 괄호 문자열입니다.
만약 A가 올바른 괄호 문자열이라면, “(A)”, “[A]”, “{A}”도 올바른 괄호 문자열입니다. 예를 들어 “[ ]”가 올바른 괄호 문자열이므로, “( [ ] )”도 올바른 괄호 문자열입니다.

만약 A, B가 올바른 괄호 문자열이라면, AB도 올바른 괄호 문자열입니다. 예를 들어 “{ }”와 “( [ ] )”가 올바른 괄호 문자열이므로, “{ } ( [ ] )”도 올바른 괄호 문자열입니다.

대괄호, 중괄호, 그리고 소괄호로 이루어진 문자열 s가 매개변수로 주어집니다. 이 s를 왼쪽으로 x (0 ≤ x < (s의 길이)) 칸만큼 회전시켰을 때 s가 올바른 괄호 문자열이 되게 하는 x의 개수를 반환하는 solution( ) 함수를 완성하세요.

• 제약조건
  s의 길이는 1 이상 1,000 이하입니다.

• 입출력의 예
  

def solution(s):
  answer = 0
  n = len(s)
  for i in range(n):
    stack = [ ]
    for j in range(n):
      # 괄호 문자열을 회전시키면서 참조
      c = s[(i + j) % n]
      if c == "(" or c == "[" or c == "{":  # 열린 괄호는 푸시
        stack.append(c)
      else:
        if not stack:  # 짝이 맞지 않는 경우
          break

        # 닫힌 괄호는 스택의 top과 짝이 맞는지 비교
        if c == ")" and stack[-1] == "(":
           stack.pop( ) 
        elif c == "]" and stack[-1] == "[":
           stack.pop( ) 
        elif c == "}" and stack[-1] == "{":
           stack.pop( ) 
        else:
             break
    else:  # for문이 break에 의해 끝나지 않고 끝까지 수행된 경우
      if not stack:
        answer += 1
  return answer