[백준 - python] 10828번: 스택

문제 링크

10828번: 스택

문제 풀이

from sys import stdin

input = stdin.readline

n = int(input())
stack = []

for _ in range(n):
    command = input().rstrip().split()
    
    match command[0]:
        case 'push':
            stack.append(int(command[1]))
        case 'pop':
            if len(stack) == 0:
                print(-1)
            else:
                print(stack.pop())
        case 'size':
            print(len(stack))
        case 'empty':
            if len(stack) == 0:
                print(1)
            else:
                print(0)
        case 'top':
            if len(stack) == 0:
                print(-1)
            else:
                print(stack[-1])

10828 스택

*key point: 파이썬 최신 문법 중 하나인 match-case문을 이용한다.