[프로그래머스/Python] 올바른 괄호

문제 설명

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

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

제한사항

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

입출력 예

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

입출력 예 설명

입출력 예 #1,2,3,4
문제의 예시와 같습니다.

---

풀이
for loop를 돌면서 현재 원소가 열린 괄호이면 stack에 넣고, 현재 원소가 닫는 괄호이면 stack의 길이를 확인한다
만약 stack이 비어있다면 짝을 지을 수 없다는 뜻이므로 즉시 False를 리턴하고 stack이 비어있지 않다면 stack에 있는 열린 괄호를 pop해준다(짝이 지어졌음을 의미)
그리고 for 문을 다 돌았는데도 stack에 남은 괄호가 있다면 짝이 맞지 않는 것을 의미하므로
False를 리턴해준다

코드

def solution(s):
    stack = []
    
    for elem in s:
        if elem == "(":
            stack.append(elem)
        else:
            if not len(stack):
                return False
            stack.pop()
    return not stack