백준 10828 파이썬

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

10828번: 스택

 

 

import sys

def push(x):
    blank.append(x)

def pop():
    if not blank:
        return -1
    else:
        return blank.pop()

def size():
    return len(blank)

def empty():
    if blank:
        return 0
    else:
        return 1

def top():
    return blank[-1] if blank else -1

blank = []
input_num = int(sys.stdin.readline())

for _ in range(input_num):
    input_data = sys.stdin.readline().rstrip().split()

    order = input_data[0]

    if order == "push":
        push(input_data[1])
    elif order == "pop":
        print(pop())
    elif order == "size":
        print(size())
    elif order == "empty":
        print(empty())
    elif order == "top":
        print(top())

파이썬의 기본 함수들을 쓰면 쉽다. 

import sys를 해주어, sys.stdin.readline()을 사용하여 입력 시간을 최소화 해야한다.