알고리즘/백준
[백준 / Python] 10828 스택 | 초코더
cloud_nice
2020. 1. 26. 16:40
https://www.acmicpc.net/problem/10828
10828번: 스택
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
www.acmicpc.net
import sys
n=int(input())
do=[]
stack=[]
for _ in range(n):
do.append(sys.stdin.readline().split())
print(do)
for i in range(len(do)):
if do[i][0]== 'push':
stack.append(do[i][1])
elif do[i][0]=='pop':
if len(stack):
#가장 위의 정수를 빼라고 했으므로 가장 마지막에 추가된 요소를 빼면된다.
print(stack.pop(-1))
else:
print(-1)
elif do[i][0] == 'size':
print(len(stack))
elif do[i][0] == 'empty':
if len(stack):
print(0)
else:
print(1)
elif do[i][0] == 'top':
if len(stack):
print(stack[-1])
else:
print(-1)