코딩테스트/Python

[Level 2] 괄호 회전하기

eunki 2021. 10. 25. 20:13
728x90

[문제]

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

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

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

 

 

[제한사항]

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

 

 

 

[코드]

def solution(s):
    sum = 0
    pl = {'(':')', '[':']', '{':'}'}
    
    for i in range(len(s)):
        s = s[1:] + s[:1]
        in_pl = []
        flag = True
        
        for st in s:
            if st in pl.keys():
                in_pl.append(pl[st])
            else:
                if len(in_pl) != 0:
                    if st == in_pl[-1]:
                        in_pl.pop()
                    else:
                        flag = False
                        break
                else:
                    flag = False
                    break

        if flag:
            if len(in_pl) == 0:
                sum += 1
    
    return sum
728x90