백준 Python 28278 스택 2

문제 링크

• list로 간단하게 스택을 구현할 수 있다.

import sys

stack = []

n = int(sys.stdin.readline())


for _ in range(n):
    
    command = sys.stdin.readline().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)