[백준/Python] 10828번 - 스택

import sys
input = sys.stdin.readline

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

스택에 대한 아주 기초적인 알고리즘 문제다.

 

 

Learned

• print문 안에서 조건문을 쓰면 특정 조건이 만족했을 때 값을 print 해준다