[python3] 10828번 스택

push 명령은 append() 함수를 이용했다.

pop 명령은 try, except 문을 이용하여 오류가 발생했을 경우 -1, 정상이면 그 값을 출력하도록 하였다.

size 명령은 len() 함수를 이용했다.

empty 명령 같은 경우에도 리스트의 길이가 0이면 1 아닐 경우 0을 출력하는 방식이다.

top 명령은 인덱싱을 이용해 출력했다.

문제풀이

J = int(input())

stack = []

for a in range(J):
	N = input().split()
	if N[0] == "push":
		stack.append(int(N[1]))
	elif N[0] == "pop":
		try:
			print(stack.pop())
		except:
			print(-1)
			continue
	elif N[0] == "size":
		print(len(stack)) 
	elif N[0] == "empty":
		if len(stack) == 0:
			print(1)
		else:
			print(0)
	elif N[0] == "top":
		if len(stack) == 0:
			print(-1)
			continue
		print(stack[-1])