프로그래머스: 올바른 괄호

문제

코딩테스트 연습 - 올바른 괄호

 

문제 풀이

 문자열이 주어질 때, 올바른 괄호인지 판단하여 true, false를 반환하는 문제이다. 예를 들어 `(())`는 올바른 괄호이고, `(()`는 올바르지 않은 괄호이다. 문제를 해결하기 위해서는 다음과 같은 로직에 따라 처리하면 된다.

 

• stack push. : 현재 문자열이 열린 괄호
• stack pop : stack에 값이 있고, 현재 문자열이 닫힌 괄호
• 이외의 경우는 올바른 괄호가 아니므로, False를 반환한다.
• 만약 문자열을 다 순회하였는데, stack에 괄호가 있다면 열린 괄호로 끝나는 케이스이다.

 

 

코드

def solution(s):
    brackets = {'(': ')'}
    '''
    괄호가 추가 된다면
    brackets = {'{': '}', '[': ']', '(': ')'}
    '''
    stack = []

    for string in s:
        if string in brackets.keys():
            stack.append(string)
        elif string in brackets.values():
            if stack and string == brackets[stack[-1]]:
                stack.pop()
            else:
                return False
    # stack에 괄호가 있는 경우, 여는 괄호의 짝이 없는 경우이다.
    return True if not stack else False

 이 문제는 간단하여 문자열에 괄호만 주어지지만, 만약 괄호의 종류 `()`, `[]`, `{}`가 많고 문자와 괄호가 섞여 들어온다면 위와 같은 로직으로 풀어야 한다. 또한 딕셔너리를 활용하여 여는 괄호에 따라, 닫는 괄호를 구분할 수 있다.