정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 다섯 가지이다.
push
일 때만 공백으로 나뉘기 때문에, push
일 때 공백으로 분리해 숫자 부분만 int
형변환하여 넣었다. 시간 제한 때문에
sys.stdin.readline()
으로 입력받았는데, 이 때str
형으로 받아오면서 개행공백이 생긴다. 그러므로rstrip()
으로 공백을 제거해주어야 된다.
import sys
n = int(sys.stdin.readline())
stack = []
for i in range(n):
cmd = sys.stdin.readline().rstrip()
if cmd == 'pop':
if stack:
print(stack[-1])
stack.pop()
else: print(-1)
elif cmd == 'size':
print(len(stack))
elif cmd == 'empty':
if stack: print(0)
else: print(1)
elif cmd == 'top':
if stack:
print(stack[-1])
else:
print(-1)
else: # push X
push_cmd, num = cmd.split()
stack.append(int(num))