[백준 - python] 28278번: 스택 2

문제 링크

28278번: 스택 2

문제 풀이

import sys
input = sys.stdin.readline

stack = []

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

28278 입출력 예시

*key point: 후입선출을 특징으로 가지는 스택 자료구조를 이해할 수 있는 기본 예제이다. 주어진 각 명령을 수행할 수 있도록 조건문을 만들어주면된다.