백준. 10828번. 스택 파이썬 풀이

백준. 10828번. 스택 파이썬 풀이

문제링크 https://www.acmicpc.net/problem/10828

그냥 input을 사용하면 시간초과가 떠서 sys 라이브러리를 사용하여 풀었다

import sys
input = sys.stdin.readline

n = int(input())

array = []

for _ in range(n):
    command = input().split()

    if "push" in command[0]:
        array.append(command[1])
    elif "top" in command[0]:
        if not array:
            print(-1)
        else:
            print(array[-1])
    elif "size" in command[0]:
        print(len(array))
    elif "empty" in command[0]:
        if not array:
            print(1)
        else:
            print(0)
    elif "pop" in command[0]:
        if not array:
            print(-1)
        else:
            print(array.pop())