티끌모아 태산

괄호 회전하기 본문

프로그래머스/Level 2

괄호 회전하기

goldpig 2024. 5. 21. 12:30
728x90

문제 설명

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

  • (), [], {} 는 모두 올바른 괄호 문자열입니다.
  • 만약 A가 올바른 괄호 문자열이라면, (A), [A], {A} 도 올바른 괄호 문자열입니다. 예를 들어, [] 가 올바른 괄호 문자열이므로, ([]) 도 올바른 괄호 문자열입니다.
  • 만약 A, B가 올바른 괄호 문자열이라면, AB 도 올바른 괄호 문자열입니다. 예를 들어, {}  ([]) 가 올바른 괄호 문자열이므로, {}([]) 도 올바른 괄호 문자열입니다.

대괄호, 중괄호, 그리고 소괄호로 이루어진 문자열 s가 매개변수로 주어집니다. 이 s를 왼쪽으로 x (0 ≤ x < (s의 길이)) 칸만큼 회전시켰을 때 s가 올바른 괄호 문자열이 되게 하는 x의 개수를 return 하도록 solution 함수를 완성해주세요.

❗제한사항

  • s의 길이는 1 이상 1,000 이하입니다.

핵심 아이디어

해당 문제는 스택을 활용해서 풀 수 있다. 

  1. 만약 스택이 비어있다면, 해당 글자를 스택에 저장한다.
  2. 스택이 비어있지 않다면, 스택의 마지막에 저장되어 있는 괄호와 현재 문자를 검사한다. 매칭이 되면 pop()
  3. s 검사 후, 스택이 비어 있다면  +1
  4. s 첫글자를 마지막 글자로 옮김

코드 구현

def solution(s):
    answer = 0
    s = list(s)
    for i in range(len(s)):
        stack = []
        for j in s:
            if len(stack) > 0:
                if stack[-1] == '[' and j == ']': stack.pop()
                elif stack[-1] == '(' and j == ')': stack.pop()
                elif stack[-1] == '{' and j == '}': stack.pop()
                else:
                    stack.append(j)
            else:
                stack.append(j)
        if len(stack) == 0:
            answer += 1
        s.append(s.pop(0))
    return answer
def solution(s):
    answer = 0
    s = list(s)
    for _ in range(len(s)):
        stack = [] # 스택 초기화
        for i in range(len(s)):
            if len(stack) > 0: # 스택이 비어있지 않다면
                if stack[-1] == '[' and s[i] == ']': stack.pop()
                elif stack[-1] == '{' and s[i] == '}': stack.pop()
                elif stack[-1] == '(' and s[i] == ')': stack.pop()
                else:
                    stack.append(s[i])
            else: # 스택이 비어 있다면
                stack.append(s[i])
        if len(stack) == 0:
            answer += 1
        s.append(s.pop(0))
        
    return answer
728x90

'프로그래머스 > Level 2' 카테고리의 다른 글

할인 행사  (1) 2024.05.29
⭐귤 고르기  (0) 2024.05.21
점프와 순간 이동  (0) 2024.05.20