10828번: 스택
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지
www.acmicpc.net
#스택
import sys
n = int(sys.stdin.readline())
stack = []
result = []
for _ in range(n):
cmd_input = sys.stdin.readline().strip().split(' ')
cmd = cmd_input[0]
if cmd == "push":
stack.append(int(cmd_input[1]))
elif cmd == "pop":
if len(stack) == 0:
result.append(-1)
else:
result.append(stack.pop())
elif cmd == "size":
result.append(len(stack))
elif cmd == "empty":
if len(stack) == 0:
result.append(1)
else:
result.append(0)
elif cmd == "top":
if len(stack) == 0:
result.append(-1)
else:
result.append(stack[len(stack)-1])
print(*result, sep = "\n")조건을 제대로 안주어서 런타임 에러(indexerror)를 보기도 했다 ㅎㅎ문제를 제대로 읽자
반응형