[알고리즘] Programmers 괄호 회전하기 #Python

[Programmers] 괄호 회전하기 바로가기

📍 문제 설명

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

• (), [], {} 는 모두 올바른 괄호 문자열입니다.
• 만약 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 해야 합니다.

---

📍 풀이

• 스택(stack) 자료구조를 이용하여 주어진 문자열이 올바른 괄호 문자열인지 확인하는 함수를 작성한다.
• 슬라이스를 이용하여 주어진 문자열 s 를 회전시킨다.
• 회전을 통해 생성할 수 있는 문자열 s 의 모든 경우를 올바른 괄호 문자열인지 확인할 수 있는 함수에 인자로 전달하여 반환받은 값을 answer 에 추가한다.

✍ 코드

def func(s):
    Stack = []
    for c in s:
        if c in ('[','(','{'): Stack.append(c)
        else:
            if not Stack: return 0
            if c == ']' and Stack[-1] == '[': Stack.pop()
            if c == ')' and Stack[-1] == '(': Stack.pop()
            if c == '}' and Stack[-1] == '{': Stack.pop()
    return 1 if not Stack else 0    

def solution(s):
    answer = 0
    for _ in range(len(s)):
        s = s[1:] + s[0]
        answer += func(s)  
    return answer