백준 알고리즘 10828번

백준링크

자료구조 스택 활용하기

파이썬

import sys
input = sys.stdin.readline

stack = []
n = int(input())
for i in range(n):
    a= input().split()
    s = a[0]
    
    if s == 'push':
        b = a[1]
        stack.append(b)
    
    elif s == 'pop':
        if len(stack) == 0:
            print(-1)
        else:
            print(stack.pop())
 
    elif s == 'size':
        print(len(stack))
        
    elif s == "empty":
        if len(stack)== 0:
            print(1)
        else:
            print(0)
            
    elif s == 'top':
        if len(stack) == 0:
            print(-1)
        else:
            print(stack[-1])