[BOJ / Python] 10828 스택

문제

백준 10828 스택

코드

import sys
N = int(sys.stdin.readline())

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

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

파이썬의 경우 별도의 stack 구조를 제공하지 않는다.
따라서 list와 append(), pop() 함수를 이용하여 stack을 구현할 수 있다.