[백준] 10828: 스택 (python 파이썬)

10828번: 스택

기본적인 스택을 구현하는 문제였다.

import sys
input = sys.stdin.readline

n = int(input())
stack = list()

for i in range(n):
    operation = input().split()
    
    if operation[0] == "push":
        stack.append(operation[1])
    elif operation[0] == "pop":
        if len(stack) == 0:
            print(-1)
        else:
            print(stack.pop()) 
    elif operation[0] == "size":
        print(len(stack))
    elif operation[0] == "empty":
        if len(stack) == 0:
            print(1)
        else:
            print(0)
    else:
        if len(stack) == 0:
            print(-1)
        else:
            print(stack[-1])