[Python] 괄호 회전하기 (프로그래머스 Lv2 파이썬)

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 이하입니다.

---

입출력 예sresult

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

---

입출력 예 설명

입출력 예 #1

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

xs를 왼쪽으로 x칸만큼 회전올바른 괄호 문자열?

0	"[](){}"	O
1	"](){}["	X
2	"(){}[]"	O
3	"){}[]("	X
4	"{}[]()"	O
5	"}[](){"	X

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

입출력 예 #2

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

xs를 왼쪽으로 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 해야 합니다.

 

 

문제 이해

- 괄호(백준), 괄호(프로그래머스)의 업그레이드 버전입니다. 3종류의 괄호르 사용하며, 단순히 올바른 괄호인지 판단하는 것 뿐만 아니라, 괄호를 회전하는 알고리즘 또한 요구합니다.

- 괄호를 회전하는 부분은 큐(덱)으로, 올바른 괄호인지 확인하는 부부은 스택으로 해결했습니다.

 

문제 풀이

- 입력받은 문자열 S를 큐에 넣습니다.

- S의 길이만큼 반복문을 돌며, i가 0인 경우엔 회전하지 않고 이외의 경우엔 회전합니다.

- "회전한다." 의 의미는 큐에서 왼쪽에서 뽑은 것을 오른쪽으로 넣는 과정입니다.(popleft하여 append)

- 회전한 큐를 바탕으로 올바른 괄호인지 판단합니다.

 

- 만약 스택이 비어있고, 닫는 괄호이면 올바른 괄호일 수 없으므로 right(=True)변수를 False로 변경 후 종료합니다.

- 만약 여는 괄호이면 스택에 넣습니다.

- 스택이 존재하는 경우에 닫는 괄호라면 스택의 마지막 요소를 확인합니다. 마지막 요소와 j가 짝을 이루는 괄호면 스택에 있는 여는 괄호를 제거합니다.

 

- 최종적으로 stack이 존재하지 않을때(아예 안들어갔거나, 올바른 괄호일때) right가 False이면(올바르지 않은 괄호) 패스하고, 올바른 괄호이면 결과에 +1 해줍니다.

 

코드

#프로그래머스 76502 괄호 회전하기

from collections import deque

def solution(s):

    count = 0

    s = deque(s)

    for i in range(len(s)):

        if i != 0:

            p = s.popleft()

            s.append(p)

 

        stack = []

        right = True

 

        for j in s:

            if not stack and j in [')', '}', ']']:

                right = False

                break

            elif j in ['(','[','{']:

                stack.append(j)

            elif stack:

                if j == ')' and stack[-1] == '(':

                    stack.pop()

                elif j == ']' and stack[-1] == '[':

                    stack.pop()

                elif j == '}' and stack[-1] == '{':

                    stack.pop()

                   

        if not stack:

            if right == False:

                continue

            else:

                count += 1

    return count

 

print(solution("[](){}"))

print(solution("}]()[{"))

print(solution("[)(]"))

print(solution("}}}"))