Baekjoon
[백준 10828] 스택
h32j00
2021. 2. 12. 00:15
문제
10828번: 스택
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지
www.acmicpc.net
처음 작성한 코드
import sys
n = int(sys.stdin.readline())
stack = []
for i in range(n):
tmp = sys.stdin.readline().split()
if tmp[0] == "push":
stack.append(tmp[1])
elif tmp[0] == "pop":
print(stack.pop()) if stack else print(-1)
elif tmp[0] == "size":
print(len(stack))
elif tmp[0] == "empty":
print(0) if stack else print(1)
elif tmp[0] == "top":
print(stack[-1]) if stack else print(-1)