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

문제 정보

제목 : 올바른 괄호

난이도 : 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,2,3,4
문제의 예시와 같습니다.

---

나의 풀이

괄호를 담을 리스트 a를 생성했습니다. 문자열 s를 반복하면서 if문을 사용하여 i원소가 '(' 이면 리스트 a에 추가해줍니다. i원소가 ')'이면 if문을 사용해 a의 길이가 0이면 False를 리턴하고 아니면 a리스트에서 pop()을 실행해줍니다. 반복이 끝나면 a에 원소가 남았는지 확인합니다. 만약 남아있다면 올바르지 않은 괄호이기 때문에 False를 리턴해줍니다. 이 모든 조건에 리턴되지 않았다면 올바른 괄호이므로 True를 리턴해줍니다.

 

코드

def solution(s):
    answer = True
    a = [];
    
    for i in s :
        if ( i == '('):
            a.append(i)
        elif ( i == ')'):
            if ( len(a) == 0):
                return False
            else :
                a.pop(len(a)-1)

    if (len(a) != 0 ):
        return False

    return True