[python] 백준 28278 스택 2

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

28278번: 스택 2

 

 

문제 풀이

• 기본적인 스택 사용에 관한 문제이다.
• 다만 문제는 시간이 걸릴 수 있다는 것이다. ($N\log{N}$)
• sys.stdin.readline으로 속도를 신경써야한다.

 

CODE

import sys
input = sys.stdin.readline
def solution(A:list):
    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):
            print(0)
        else:
            print(1)
    else:
        if len(stack):
            print(stack[-1])
        else:
            print(-1)
        

if __name__=='__main__':
    n = int(input())
    stack = []
    for _ in range(n):
        A = list(map(int, input().split()))
        solution(A)