백준 (boj) 10828 파이썬 - 스택

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

10828번: 스택

 

 

딱히 어려웠던건 없었지만 sys.readline.rstrip() 을 사용하지 않고 그냥 input()으로 받았더니 시간초과가 발생하였다.

import sys
input = sys.stdin.readline
N = int(input())
stack = []
for i in range(N):

    command = input().rstrip()
    
    if " " in command :    #입력 문자열에 공백이 포함된건 push밖에 없다. 
        c,n = command.split()
        stack.append(int(n))
    elif command == 'pop' : 
        if len(stack) > 0 : print(stack.pop(-1))
        else : print(-1)

    elif command == 'size' : print(len(stack))
    elif command == 'empty' : print(1 if len(stack)==0 else 0)
    elif command == 'top' :
        if len(stack) > 0 : print(stack[-1])
        else : print(-1)