<백준 문제풀이: 10828번> 파이썬 - 스택

간단한 수학 문제인데, Hackerrank에서 봤었던 문제라 간단하게 풀 수 있었다.

import sys

N = int(sys.stdin.readline())
nums = []

for i in range(N):
    a = sys.stdin.readline().strip()
    if a.split( )[0] == 'push':
        nums.append(a.split( )[1])
    elif a == 'top':
        if len(nums) == 0:
            print(-1)
        else:
            print(nums[-1])
    elif a == 'size':
        print(len(nums))
    elif a == 'empty':
        if len(nums) == 0:
            print(1)
        else:
            print(0)
    elif a == 'pop':
        if len(nums) == 0:
            print(-1)
        else:
            print(nums[-1])
            del nums[-1]