[프로그래머스] 올바른 괄호 (스택/큐 Lv.2) - Python

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12909

프로그래머스

문제설명

괄호가 바르게 짝지어졌다는 것은 '(' 문자로 열렸으면 반드시 짝지어서 ')' 문자로 닫혀야 한다는 뜻입니다. 예를 들어

"()()" 또는 "(())()" 는 올바른 괄호입니다.
")()(" 또는 "(()(" 는 올바르지 않은 괄호입니다.
'(' 또는 ')' 로만 이루어진 문자열 s가 주어졌을 때, 문자열 s가 올바른 괄호이면 true를 return 하고, 올바르지 않은 괄호이면 false를 return 하는 solution 함수를 완성해 주세요.

제한사항

• 문자열 s의 길이 : 100,000 이하의 자연수
• 문자열 s는 '(' 또는 ')' 로만 이루어져 있습니다.

입출력 예

s	answer
"()()"	true
"(())()"	true
")()("	false
"(()("	false

풀이

접근

1. 문자열 s 를 차례대로 검사
2. "(" 가 나오면 stack에 쌓기
3. ")" 가 나오면 stack 가장 위에부터 빼기
4. ")" 가 나왔는데 쌓인 "("가 없으면 False
5. 문자열을 다 검사했는데 stack에 남아있는 게 있으면 짝이 안 맞는 것이므로 False

풀이 코드

def solution(s):
    answer = True
    stack = []
    for char in s:
        if char == "(":
            stack.append(char)
        else:
            if not stack:
                return False
            top = stack.pop()
    if stack:
        return False
    return True