[백준] 10828번: 스택 - python

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

10828번: 스택

 

stack class를 정의하여 stack의 기능/명령을 구현함 

sys를 쓰지 않으니 시간초과가 났음 import sys로 시간초과 해결 

 

정답코드:

import sys
input = sys.stdin.readline

class Stack:
  def __init__(self):
    self.data=[]
  def empty(self):
    if len(self.data) == 0:
      return 1
    else:
      return 0
  def size(self):
    return len(self.data)
  def push(self, X):
    self.data.append(X)
  def pop(self):
    if self.size() == 0:
      return -1
    else:
      return self.data.pop()
  def top(self):
    if self.size() == 0:
      return -1
    else:
      return self.data[-1]

num = int(input())
stack = Stack()
for _ in range(num):
  cmd = input().split()

  op = cmd[0]
  if op == 'push':
    stack.push(cmd[1])
  elif op == 'empty':
    print(stack.empty())
  elif op == 'pop':
    print(stack.pop())
  elif op == 'size':
    print(stack.size())
  elif op == 'top':
    print(stack.top())
  else:
    print('err')