문제

10828번: 스택

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지

www.acmicpc.net

코드

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인덱스를 주어서 가장 최근에 들어간 부분이 나오도록 설정해주었다.