백준 10828 - 스택 (파이썬)

Answer

import sys
n = int(sys.stdin.readline())
stack = []

for _ in range(n):
    order = sys.stdin.readline().split()

    if order[0] == 'push':
        num = int(order[1])
        stack.append(num)
    if order[0] == 'pop':
        if stack:
            print(stack.pop())
        else:
            print(-1)
    if order[0] == 'size':
        print(len(stack))
    if order[0] == 'empty':
        if stack:
            print(0)
        else:
            print(1)
    if order[0] == 'top':
        if stack:
            print(stack[-1])
        else:
            print(-1)