[Python] 백준 10773

eun·2024년 1월 17일

문제 10773

먼저 문제를 이해해보자

  • 첫째줄에는 정수 k가 주어지며 이후 k개의 줄에 정수가 하나씩 주어진다.
  • 정수가 0이면 pop 하고, 0이 아니면 push 한다.
# 10773
import sys
n = int(input())

stack = []

for i in range(n):
    num = int(sys.stdin.readline())
    if num != 0:
        stack.append(num)
    else:
        stack.pop()
        
s = 0

print(sum(stack))

위 문제는 스택의 push, pop 개념을 이용하여 쉽게 풀 수 있었다.
처음에는 sys.stdin.readline() 사용 대신 input()을 사용하여 정수를 입력받았는데 실행 시간이 너무 길게 나와 sys를 이용하는 것으로 바꾸었다.

0개의 댓글