stack을 명령어에 따라 처리하도록 구현하는 문제
https://www.acmicpc.net/problem/10828
stack을 클래스로 구현하지 않고 반복문을 통해 명령어 처리
import sys
def stack_sim(n) :
stack = []
for i in range(n) :
order = sys.stdin.readline().split()
if order[0] == "empty" :
if len(stack) == 0 : print(1)
else : print(0)
elif order[0] == "push" :
stack.append(int(order[1]))
elif order[0] == "pop" :
if len(stack) == 0 : print(-1)
else : print(stack.pop())
elif order[0] == "size" :
print(len(stack))
elif order[0] == "top" :
if len(stack) == 0 :
print(-1)
else : print(stack[-1])
stack_sim(int(sys.stdin.readline()))
n개의 입력값에 대해 처리하기 때문에 O(n)