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

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 s를 왼쪽으로 x칸만큼 회전 올바른 괄호 문자열?

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

입출력 예 #3

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

입출력 예 #4

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

---

※ 공지 - 2021년 4월 16일 테스트케이스가 추가되었습니다.

 

풀이

입출력 예에서 보여지듯 가능한 문자열을 모두 만든 뒤 올바른 괄호 문자열을 확인해봅시다. 문자열의 회전은 간단합니다. 문자열을 특정 인덱스를 기준으로 잘라 두 덩어리의 순서를 바꿔주면 됩니다.

for i in range(len(s)):
    string = s[i:] + s[:i]

위 코드를 통해 가능한 모든 문자열을 얻을 수 있습니다.

올바른 괄호 문자열을 찾는 방법은 많은 문제에서 다루고 있습니다. [, {, ( 이 나오면 스택에 집어넣고, ), }, ] 이 나왔을 땐 스택 최상단과 비교해서 짝이 맞을 경우에만 pop 해줍니다. 이외의 경우에는 잘못된 문자열이므로 루프를 중단합니다.

루프를 중단했을 때 스택이 비어있다면 올바른 괄호 문자열입니다. 아니라면 "({["과 같은 경우입니다.

pair = {']': '[', ')': '(', '}': '{'}

def solution(s):
    answer = 0
    for i in range(len(s)):
        string = s[i:] + s[:i]
        stack = []
        for char in string:
            if char in "[({":
                stack.append(char)
            elif stack != [] and pair[char] == stack[-1]:
                stack.pop()
            else:
                break
        else:
            if stack == []:
                answer += 1
    return answer

pair를 찾기 위해 처음에는 아스키 코드 값을 사용하였습니다. 괄호 문자의 아스키 코드는 서로 붙어있을 것이란 가정이었으나, 실제 비교 결과 {, }, [, ]는 서로 2만큼 떨어져있어 대신 딕셔너리를 사용하여 해결했습니다.