[Python] 백준 10828번(스택) 문제 풀이

 

import sys

stack = []
for i in range(int(sys.stdin.readline())):
    command = sys.stdin.readline()
    if "push" in command:
        stack.append(int(command.split()[1]))
        continue
    if command == "pop\n":
        try:
            print(stack.pop())
        except:
            print(-1)
    elif command == "size\n":
        print(len(stack))
    elif command == "empty\n":
        if stack == []:
            print(1)
        else:
            print(0)
    else:
        try:
            print(stack[-1])
        except:
            print(-1)