[백준] 10828번 스택(with python)

🔅코딩테스트 공부🔅/❗백준

[백준] 10828번 스택(with python)

윤무무 2023. 1. 29. 22:11

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

10828번: 스택

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

www.acmicpc.net

1. 내가 작성한 코드

import sys

n = int(sys.stdin.readline())

stack = []

for i in range(n):
  word = sys.stdin.readline().split()

  if word[0] == 'push':
    stack.append(word[1])

  elif word[0] == 'pop':
    if len(stack) > 0:
      print(stack.pop())
    else:
      print(-1)

  elif word[0] == 'size':
    print(len(stack))

  elif word[0] == 'empty':
    if len(stack) == 0:
      print(1)
    else:
      print(0)

  elif word[0] == 'top':
    if len(stack) > 0:
      print(stack[-1])
    else:
      print(-1)

아~ 함수 이용해서 풀려고 했는데 자꾸 오류떠서 그냥 if문으로 풀었다.

담에 함수로 다시 해봐야지