괄호 회전하기

Ja L·2022년 12월 30일
0

[코테] Python

목록 보기
7/9

문제

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

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

문제 풀이

def solution(s):
    ls = list(s) # ['[', ']', '(', ')', '{', '}']
    count = 0
    for _ in range(len(s)): # s 개수만큼 밀기 반복
        stack = []
        for i in range(len(ls)): # ls 개수 횟수만큼 반복
            if len(stack) == 0: # 스택이 비어있으면
                stack.append(ls[i]) # 스택에 값을 넣어라
            else:  # 스택 마지막 값이 열린거고 짝이 맞다면 스택에서 삭제하고 / 그게아니라면 한번 더 값을 넣어라
                if stack[-1] == "(" and ls[i] == ")":
                    stack.pop()
                elif stack[-1] == "{" and ls[i] == "}":
                    stack.pop()
                elif stack[-1] == "[" and ls[i] == "]":
                    stack.pop()
                else:
                    stack.append(ls[i])
        if len(stack) == 0: # 스택이 비어있다면 (괄호가 모두 짝이 잘 맞았던 것) 
            count += 1 # 카운트를 해라
        
        ls.append(ls.pop(0))
        
    return count
    
    

profile
DB Engineer

0개의 댓글