[프로그래머스] 괄호 회전하기_Python

문제 설명

 

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

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

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

 

제한사항

s의 길이는 1이상 100이하입니다

 

 

해결 코드

def solution(test):
  answer= 0
  n=len(test)
  for i in range(n):
    stack = [ ]
    for j in range(n): # test를 회전하기 위해 이중 for문사용
      c = test[(i+j)%n]   
      if c == "(" or c == "{" or c == "[":
        stack.append(c) #스택에 푸시
      else:
        if not stack: #짝이 맞지 않는 경우는 break
          break
        if c == ")" and stack[-1] == "(":
          stack.pop()
        elif c == "]" and stack[-1] == "[":
          stack.pop()
        elif c == "}" and stack[-1] == "{":
          stack.pop()
        else:
          break
    else: # break에 걸리지 않고 끝까지 수행한 경우
        if not stack:
          answer +=1        
  return answer