알고리즘 풀이/Python

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

모남(monam2) 2023. 9. 7. 17:08

https://school.programmers.co.kr/learn/courses/30/lessons/76502

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

문제 설명

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

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