[프로그래머스] 괄호 회전하기 - 파이썬

링크: 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 함수를 완성해주세요.

---

제한사항

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

---

입출력 예

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

---

입출력 예 설명

입출력 예 #1

• 다음 표는 "[](){}" 를 회전시킨 모습을 나타낸 것입니다.

x	s를 왼쪽으로 x칸만큼 회전	올바른 괄호 문자열?
0	"{}"	O
1	"](){}["	X
2	"(){}[]"	O
3	"){}[]("	X
4	"{}"	O
5	"}{"	X

• 올바른 괄호 문자열이 되는 x가 3개이므로, 3을 return 해야 합니다.

입출력 예 #2

• 다음 표는 "}]()[{" 를 회전시킨 모습을 나타낸 것입니다.

x	s를 왼쪽으로 x칸만큼 회전	올바른 괄호 문자열?
0	"}]()[{"	X
1	"]()[{}"	X
2	"()[{}]"	O
3	")[{}]("	X
4	"{}"	O
5	"{}]()["	X

• 올바른 괄호 문자열이 되는 x가 2개이므로, 2를 return 해야 합니다.

입출력 예 #3

• s를 어떻게 회전하더라도 올바른 괄호 문자열을 만들 수 없으므로, 0을 return 해야 합니다.

입출력 예 #4

• s를 어떻게 회전하더라도 올바른 괄호 문자열을 만들 수 없으므로, 0을 return 해야 합니다.

---

문제풀이

문제 이해하기:

목표: 대괄호, 중괄호, 그리고 소괄호로 이루어진 문자열 s 가 주어질 때, 이 문자열 s 를 왼쪽으로 x 칸만큼 회전시켰을 때 s 가 올바른 괄호 문자열이 되게 하는 x 의 개수를 return

최종코드

def solution(s):
    # 문자열이 올바른 괄호 문자열인지 확인
    def is_valid_paren(paren_string):
        stack = []  
        matching = {')': '(', ']': '[', '}': '{'}  # 닫는 괄호에 대한 짝이 되는 여는 괄호를 매핑

        for char in paren_string:
            if char in matching.values():  # 여는 괄호라면 스택에 추가
                stack.append(char)
            elif char in matching.keys():  # 닫는 괄호라면 스택에서 마지막 여는 괄호와 짝을 맞추고
                if stack and stack[-1] == matching[char]:  # 짝이 맞으면 스택에서 제거
                    stack.pop()
                else:
                    return False  # 짝이 맞지 않으면 올바른 괄호 문자열이 아님

        return len(stack) == 0  # 성공적으로 스택을 비웠으면 모든 괄호가 짝이 맞은 것

    count = 0  # 올바른 괄호 문자열이 되는 회전의 개수를 카운트
    for i in range(len(s)):
        rotated_s = s[i:] + s[:i]  # 문자열 s를 슬라이싱으로 회전
        if is_valid_paren(rotated_s):  # 회전된 문자열이 올바른 괄호 문자열인지 확인
            count += 1  # 올바른 경우 카운트 증가
    
    return count  # 최종적으로 올바른 괄호 문자열이 되는 회전의 개수를 반환