[백준/python/10828] 스택

bej_ve·2022년 4월 12일
0

python알고리즘

목록 보기
16/46


문제링크 : 스택

import sys
input=sys.stdin.readline

n=int(input())
stack=[]
for _ in range(n):
   c=input().split()
   if c[0]=='push':
       stack.append(c[1])
   elif c[0]=='pop':
       if len(stack)==0:
           print(-1)
       else:
           print(stack.pop())
   elif c[0]=='size':
       print(len(stack))
   elif c[0]=='empty':
       if len(stack)==0:
           print(1)
       else:
           print(0)
   elif c[0]=='top':
       if len(stack)==0:
           print(-1)
       else:
           print(stack[-1])

간단한 스택을 구현하는 문제이다. 배열의 명령어를 사용해서 풀어줬다.

0개의 댓글