[Python/Programmers] 괄호 회전하기

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의 개수를 return 하도록 solution 함수를 완성해주세요.

풀이

Code

def solution(s):
    # 1. 올바른 괄호 문자열 체크 함수 정의
    def check(s) :
        # 1-1. 스택 생성
        stack = []
        # 1-2.
        while s :
            # 1-2-1. 스택이 비었을 경우
            if not stack : stack.append(s.pop(0)) # 첫 값 삽입
            # 1-2-2. 스택이 비어있지 않을 경우
            else :
                # 스택의 마지막 문자와 남은 문자열의 앞 문자가 한 쌍일 경우
                if [stack[-1], s[0]] in [['[', ']'], ['{', '}'], ['(', ')']] :
                    stack.pop()
                    s.pop(0)
                # 이외의 경우
                else :
                    stack.append(s.pop(0))
        # 1-3. 결과 출력
        return 1 if not stack else 0
    answer = 0
    # 2. 문자열 리스트화시키기
    s = list(s)
    # 3.
    for x in range(len(s)) :
        # 문자 회전
        s.append(s.pop(0))
        # 체크 함수 실행
        answer += check(s.copy())
    # 4. 결과 출력
    return answer