백준 10828번 스택 파이썬

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

10828번: 스택

 

import sys
stack = []
for i in range(int(sys.stdin.readline())):
    x = sys.stdin.readline().split()
    if x[0] == 'push':
        x[1] = int(x[1])
        stack.append(x[1])
    elif x[0] == 'pop':
        if len(stack) > 0:
            print(stack.pop())
        else:
            print('-1')
    elif x[0] == 'size':
        print(len(stack))
    elif x[0] == 'empty':
        if len(stack) == 0:
            print('1')
        else:
            print('0')
    elif x[0] == 'top':
        if len(stack) > 0:
            print(stack[len(stack)-1])
        else:
            print('-1')

시간 제한 때문에 sys 라이브러리를 이용해 입력을 받았고

무지성으로 if else 문을 이용한 코드를 짰다.

원래는 class를 활용하여 푸려고 했으나, push 1 이라는 명령어를 입력을 받아야 해서 리스트 처럼 나눠

x = sys.stdin.readline().split()

이렇게 입력을 받았다.