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

문제 설명

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

(), [], {} 는 모두 올바른 괄호 문자열입니다.
만약 A가 올바른 괄호 문자열이라면, (A), [A], {A} 도 올바른 괄호 문자열입니다. 

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

예를 들어, {} 와 ([]) 가 올바른 괄호 문자열이므로, {}([]) 도 올바른 괄호 문자열입니다.
대괄호, 중괄호, 그리고 소괄호로 이루어진 문자열 s가 매개변수로 주어집니다. 

이 s를 왼쪽으로 x (0 ≤ x < (s의 길이)) 칸만큼 회전시켰을 때 s가 올바른 괄호 문자열이 되게 하는 x의 개수를 return 하도록 solution 함수를 완성해주세요.

 

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

프로그래머스

제한 사항

s의 길이는 1 이상 1,000 이하입니다.

예제

s	result
[](){}	3
}]()[{	2
[)(]	0
}}}	0

문제 풀이

def is_valid(s):
    stack = []
    bracket = {'(':')', '{':'}', '[':']'}
    
    for char in s:
        # 여는 괄호일 경우 stack에 추가
        if char in bracket.keys():
            stack.append(char)
            
        # 닫는 괄호일 경우
        else:
            # 스택이 비어 있지 않고, 마지막 여는 괄호와 매칭되는 경우
            if stack and char == bracket[stack[-1]]:
                # 스택 내 여는 괄호 제거
                stack.pop()
            # 매칭되지 않는 경우 False 반환
            else:
                return False
            
    # 스택이 비어있으면 True 반환
    return len(stack) == 0

def solution(s):
    answer = 0
    n = len(s)
    
    for i in range(n):
        # 문자열 회전
        rotated = s[i:] + s[:i]
        if is_valid(rotated):
            answer += 1
            
    return answer

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.