[백준] 10828번 스택(파이썬)

문제

10828번: 스택

코드

from collections import deque
import sys
input = sys.stdin.readline

N = int(input())
stack = deque([])

for _ in range(N):
     order = input().strip()
   
     if "push" in order:
        o = order.split(" ")
        stack.append(o[1])

     elif order == "pop":
          if not stack:
               print(-1)
          else:
               print(stack.pop())
     
     elif order == "size":
          print(len(stack))
     
     elif order == "empty":
          if not stack:
               print(1)
          else:
               print(0)
     elif order == "top":
          if not stack:
               print(-1)
          else:
               print(stack[-1])

해설

스택을 표현하는 문제이다.

스텍을 따로 파이썬으로 사용할 함수는 없으니 deque를 사용했다. 속도가 빠를까 싶어 사용하여 없어도 무관하다.

push부분만 유의해서 in으로 받아서 따로 split() 해주었다.

top부분은 -1인덱스를 주어서 가장 최근에 들어간 부분이 나오도록 설정해주었다.