[BOJ] 10828번: 스택

■ 문제해설

각 기능별로 (push, pop, size, empty, top) 함수로 구현하였고, stack이 비어있는 경우에는 len(stack) == 0을 이용하여 조건문을 구성하였다.

---

■ 문제해답

import sys
input = sys.stdin.readline

n = int(input())
stack = []

def push(x):
  return stack.append(x)

def pop():
  if len(stack) == 0:
    return -1
  else:
    return stack.pop()

def size():
  return len(stack)

def empty():
  if len(stack) == 0:
    return 1
  else:
    return 0

def top():
  if len(stack) == 0:
    return -1
  else:
    return stack[len(stack) - 1]


for i in range(n):
  arr = input().split()
  m = arr[0]

  if m == 'push':
    push(arr[1])
  elif m == 'pop':
    print(pop())
  elif m == 'empty':
    print(empty())
  elif m == 'size':
    print(size())
  elif m == 'top':
    print(top())

---

■ 문제 출처

https://www.acmicpc.net/problem/10828

10828번: 스택