백준 28278 (python): 스택 2

import sys

n = int(input())
stack = []
for i in range(n):
    cmd = list(map(int, sys.stdin.readline().split()))
    if cmd[0] == 1:
        stack.append(cmd[1])
    elif cmd[0] == 2:
        if stack:
            print(stack.pop())
        else:
            print(-1)
    elif cmd[0] == 3:
        print(len(stack))
    elif cmd[0] == 4:
        if stack:
            print(0)
        else:
            print(1)
    else:
        if len(stack) > 0:
            print(stack[-1])
        else:
            print(-1)

 

하라는대로 하기

파이썬의 list pop(), append() 시간복잡도는 O(1)

 

💡 파이썬 리스트가 비어있는지 체크하는 방법

if stack:
        print(0)
    else:
        print(1)