[백준 10828번 문제 스택 파이썬(python)] input()은 쓰지 말자

문제

 

설명

단순히 명령문에 따라서 시행하면 되는 문제

 

코드

import sys

N = int(sys.stdin.readline())
stk = []
for _ in range(N):
    command = sys.stdin.readline().split()
    if command[0] == "push":
        stk.append(command[1])
    elif command[0] == "pop":
        if stk:
            print(stk.pop())
        else:
            print(-1)
    elif command[0] == "size":
        print(len(stk))
    elif command[0] == "empty":
        print(1 if not (stk) else 0)
    elif command[0] == "top":
        print(stk[-1] if stk else -1)

처음엔 input()으로 받아서 했었는데 시간초과가 떴고 찾아보니 sys를 사용하면 빠르게 처리할 수 있어서 통과했다