[백준] 10828번 스택 (Python)

문제

www.acmicpc.net/problem/10828

10828번: 스택

 

코드

import sys
N = int(input())
stack = []
a = []

def push(X):
    stack.append(X)

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

def size():
    return len(stack)

# false = 0, true =1

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

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

for i in range(N):
    a=sys.stdin.readline().split()
    if a[0] == 'push':
        push(int(a[1]))
    elif a[0] == 'top':
        print(top())
    elif a[0] == 'size':
        print(size())
    elif a[0] == 'empty':
        print(empty())
    elif a[0] == 'pop':
        print(pop())

 

코드설명

 

스택을 직접 구현하는 문제인데 반복문에서 input()으로 받게되면 시간초과가 떠서 sys를 import해줘서

sys.stdin.readline()으로 받아줬다.