[프로그래머스] LV2. 괄호 회전하기 (Python)

문제 설명

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

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

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

제한사항

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

---

풀이

1. 여는 괄호를 스택에 넣기
2. 닫는 괄호가 나오는 경우 여는 괄호와 일치 여부 확인 후 빼내기
3. 만약 닫는 괄호만 있는 경우를 감안하여 여는 괄호가 있는지 여부 확인

def solution(s):
    stirng_length = len(s)
    cnt = 0
    for i in range(stirng_length):
        # 문자열 재배치 (시간 多)
        strings = s[i+1:] + s[:i+1]
        # 스택 이용해서 괄호 넣었다가 빼기
        stack = []
        # 여는 괄호 없는 경우
        isOpen = False
        for string in strings:
            # 여는 괄호일 경우
            if string == '(' or string == '{' or string == '[':
                stack.append(string)
                isOpen = True
            # 닫는 괄호일 경우 스택에 들어간 여는 괄호 검사
            else:
                if len(stack) > 0:
                    if stack[-1] == '(' and string == ')':
                        stack.pop()
                    elif stack[-1] == '{' and string == '}':
                        stack.pop()
                    elif stack[-1] == '[' and string == ']':
                        stack.pop()
        if len(stack) == 0 and isOpen:
            cnt += 1
    return cnt

느낀점

스택 문제라는 것을 인지하고 풀었는데, 마지막 쯤에 모두 닫는 괄호만 있는 경우를 고려하지 못했다. 항상 반례를 명심하고 풀어야 하는 듯