BOJ 백준 28278 스택 2 풀어보기 [Python]

https://www.acmicpc.net/problem/28278

28278번: 스택 2

 

실버 4

 

 

빡구현문제 + 시간초과

import sys
x = int(sys.stdin.readline())
stack = []

for i in range(x):
    A = list(map(int, sys.stdin.readline().split()))
    if A[0] == 1:
        stack.append(A[1])
    elif A[0] == 2:
        if len(stack) > 0:
            print(stack.pop())
        else:
            print(-1)
    elif A[0] == 3:
        print(len(stack))
    elif A[0] == 4:
        if len(stack) == 0:
            print(1)
        else:
            print(0)
    elif A[0] == 5:
        if len(stack) <= 0:
            print(-1)
        else:
            print(stack[-1])