[Python] 백준 #10828 - 스택

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

10828번: 스택

 

 

풀이

import sys
input = sys.stdin.readline
n = int(input())
a = []
for _ in range(n):
    x=input().split()
    if x[0]=="push": a.append(x[1])
    elif x[0]=="pop": print(-1 if len(a)==0 else a.pop())
    elif x[0]=="size":print(len(a))
    elif x[0]=="empty":print(1 if len(a)==0 else 0)
    elif x[0]=="top": print(-1 if len(a)==0 else a[-1])

 

조건문으로 나누어서 풀어주었다.

sys.stdin.readline을 쓰지 않으면 시간초과가 난다.