(백준)10828 스택 Python

문제 풀이

import sys
stack = []
N = int(sys.stdin.readline())
for i in range(N):
    cmd = sys.stdin.readline().split()

    if cmd[0] == 'push':
        stack.append(cmd[1])
    elif cmd[0] == 'pop':
        if stack: print(stack.pop())
        else: print(-1)
    elif cmd[0] == 'size':
        print(len(stack))
    elif cmd[0] == 'empty':
        if stack: print(0)
        else: print(1)
    else:
        if stack: print(stack[-1])
        else: print(-1)

1. 문제에서 말한 것 처럼 정수를 저장하는 stack을 list로 선언해주기
2. input.split()으로 받은 명령어들을 if/elif문을 통해 분류하고 해당 명령에 따라서 처리해준 후 출력해주면 끝!