[백준 문제풀이] 28278 - 스택 2

백준 문제풀이 28278번

풀이

파이썬에서는 스택을 ‘리스트’로 쉽게 구현할 수 있다.

코드

import sys

input = sys.stdin.readline

stack = []

n = int(input())
for _ in range(n):
    command = input().split()
    
    if command[0] == '1':
        stack.append(command[1])
    elif command[0] == '2':
        if stack:
            print(stack.pop())
        else:
            print(-1)
    elif command[0] == '3':
        print(len(stack))
    elif command[0] == '4':
        if stack:
            print(0)
        else:
            print(1)
    elif command[0] == '5':
        if stack:
            print(stack[-1])
        else:
            print(-1)