백준 4949 | 균형잡힌 세상 [파이썬 Python]

4949번: 균형잡힌 세상

 

 

처음 시도

while(True):
    word=input()

    c=d=0
    if word=='.':
        print('Yes')
        break
    else:
        for w in word:
            if w=='(':
                c+=1
            if w==')':
                c-=1
            if w=='[':
                d+=1
            if w==']':
                d-=1
        if c==0 and d==0:
            print('Yes')
        else:
            print('No')

단순히 괄호의 짝만 맞추면 되는 것으로 이해하고 접근했으나,

균형이 되려면 괄호 속 괄호도 균형이 맞아야 한다는 조건이 있었음.

 

괄호 속 괄호의 균형도 파악하기 위해

리스트를 사용하여 마지막 요소와 현재 읽은 문자가 짝이 맞는지 확인하는 과정을 거침.

---

정답 코드

while(True):
    word=input()

    stack=[] # 괄호를 추가할 리스트 stack
    if word=='.': # . 이 들어오면 종료
        break

    for w in word:
        if w=='(' or w=='[': # 여는 괄호가 오면 stack에 추가
            stack.append(w)
        elif w==')': # ) 인 경우
            if len(stack)!=0 and stack[-1]=='(': # stack이 비어있지 않고, 마지막 요소가 ( 이면 pop
                stack.pop()
            else: # stack이 비어있거나 짝이 맞지 않으면 stack에 )을 추가하고 break
                stack.append(w)
                break
        elif w==']': # ] 인 경우
            if len(stack)!=0 and stack[-1]=='[': # stack이 비어있지 않고, 마지막 요소가 [ 이면 pop
                stack.pop()
            else: # stack이 비어있거나 짝이 맞지 않으면 stack에 ]을 추가하고 break
                stack.append(w)
                break

    if len(stack)==0: # stack이 비어있으면 모든 괄호들의 균형이 맞으므로 Yes 출력
        print('yes')
    else: # stack이 비어있지 않으면 괄호들의 균형이 맞지 않는 것이므로 No 출력
        print('no')