BAEKJOON_9012 "괄호" PYTHON

[백준] 9012번 Python 파이썬

 

https://www.acmicpc.net/problem/9012

9012번: 괄호

< 틀린 코드 >

 

처음에는 리스트를 만든 후 ' ( ' 를 입력받으면 리스트에 담고 ' ) ' 를 입력받으면 pop() 함수를 이용해서 리스트에서 제거해주었는데 리스트에 아무것도 없는데 ' ) ' 를 입력받으면 IndexError가 뜬다.

import sys

n = int(sys.stdin.readline())
lt = []

for i in range(n):
    a = input()
    for i in range(len(a)):
        if a[i] == '(':
            lt.append(a[i])
        elif a[i] == ')':
            if len(lt) == 0:
                break
            else:
                lt.pop()
    if len(lt) == 0:
        print('YES')
    else:
        print('NO')
    lt = []

< 정답 코드 >

 

리스트에 담지 않고 count라는 변수를 지정하여 ' ( ' 를 입력받으면 count에 1을 더해주고 ' ) '를 입력받으면 count에서 1을 빼주었다.

 

만약 count가 음수가 된다면 짝이 맞지 않기 때문에 break로 for문을 끝낸다.

import sys

n = int(sys.stdin.readline())
count = 0

for i in range(n):
    a = input()
    for i in range(len(a)):
        if a[i] == '(':
            count += 1
        elif a[i] == ')':
            count -= 1
        if count < 0:
            break
    if count == 0:
        print('YES')
    else:
        print('NO')
    count = 0

 

혼자 공부하며 올리는 블로그입니다. 틀린 내용은 댓글 남겨주시면 감사하겠습니다.