[9/12] 괄호 회전하기

이경준·2021년 9월 12일
0

코테

목록 보기
105/140
post-custom-banner

레벨2 문제

내 코드

from collections import deque

def solution(s):
    answer = 0
    
    for i in range(len(s)):
        temp = s[i:] + s[:i]
        answer += check(temp)
    
    return answer

def check(k):
    k = deque(k)
    
    arr = []
    while k:
        n = k.popleft()
        if ( n in ('[', '(', '{') ):
            arr.append(n)
        else:
            if ( n == ']' and len(arr) != 0 and arr[-1] == '[' ):
                arr.pop()
            elif ( n == ')' and len(arr) != 0 and arr[-1] == '(' ):
                arr.pop()
            elif ( n == '}' and len(arr) != 0 and arr[-1] == '{' ):
                arr.pop()
            else:
                arr.append(n)
                
    if len(arr) == 0:
        num = 1
    else:
        num = 0
    return num

로직

  • 스텍 자료구조 이용
profile
The Show Must Go On
post-custom-banner

0개의 댓글