[프로그래머스/Level2] 괄호 회전하기(월간 코드 챌린지 시즌2)

문제 설명

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

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

 

---

나의 풀이

[Python(파이썬)]

from collections import deque
def check(s):
    stack = []
    for i in s:
        if not stack:
            stack.append(i)
        elif i in ['[', '(', '{']:
            stack.append(i)
        elif (i == ']' and stack[-1] == '[') or (i == ')' and stack[-1] == '(') or (i == '}' and stack[-1] == '{'):
            del stack[-1]
        else:
            stack.append(i)

    return len(stack) == 0


def solution(s):
    answer = 0
    deq = deque(s)

    for i in range(len(s)):
        if check(deq) == True:
            answer += 1
        deq.rotate(-1)

    return answer

 

---

풀이 과정

1. 올바른 괄호 문자열 check 함수
2. rotation

 

1. 올바른 괄호 문자열 check 함수

 

def check(s):	# 올바른 괄호인지 check하는 함수 리턴 값은 t,f
    stack = []
    for i in s:
        if not stack:	# stack에 값이 비어있는 경우
            stack.append(i)
        elif i in ['[', '(', '{']:	# 열린 괄호 인 경우
            stack.append(i)
        # stack[-1]이 닫힌 괄호에 대응하는 올바른 열린 괄호 일때
        elif (i == ']' and stack[-1] == '[') or (i == ')' and stack[-1] == '(') or (i == '}' and stack[-1] == '{'):
            del stack[-1]
        else:
            stack.append(i)

    return len(stack) == 0

 

 

 

2. rotation

 

문자열을 왼쪽으로 회전 시키기 위해서 deque rotate를 사용했다. 

 

deq = deque(s)	# 문자열 s로 deque 생성

    for i in range(len(s)):
        if check(deq) == True:
            answer += 1
        deq.rotate(-1)	# 왼쪽으로 1만큼 회전

 

• deque에서 item들을 값 만큼 회전 해주는 메소드 - rotate(값)

# deque 생성
>>> deq = deque(['A','B','C','D','E'])
>>> deq
deque(['A', 'B', 'C', 'D', 'E'])

# 오른쪽으로 1 회전
>>> deq.rotate(1)
>>> deq
deque(['E', 'A', 'B', 'C', 'D'])

# 왼쪽으로 1 회전
>>> deq.rotate(-1)
>>> deq
deque(['A', 'B', 'C', 'D', 'E'])
>>> deq.rotate(3)
>>> deq
deque(['C', 'D', 'E', 'A', 'B'])
>>> deq.rotate(-3)
>>> deq
deque(['A', 'B', 'C', 'D', 'E'])

 

[자료구조] 데큐(deque; double-ended queue)

 

 

---

문제 출처

코딩테스트 연습 - 괄호 회전하기