[백준] 10828번 스택 / 파이썬(python)

# 문제 링크

www.acmicpc.net/problem/10828

10828번: 스택

# Soultion

시간초과 문제가 발생하므로, 5줄에서 input() 대신 sys.stdin.readline()을 이용하여 읽어온다.파이썬에서는 리스트를 이용하여 스택으로 사용이 가능하다.

import sys
n = int(input())
stack = []
for _ in range(n):
    cmd = list(map(str,sys.stdin.readline().split()))
    if cmd[0]=='push':
        stack.append(int(cmd[1]))
    elif cmd[0]=='pop':
        if stack:
            print(stack.pop())
        else:
            print("-1")
    elif cmd[0]=='size':
        print(len(stack))
    elif cmd[0]=='empty':
        if len(stack)==0:
            print(1)
        else:
            print(0)
    elif cmd[0]=='top':
        if stack:
            print(stack[-1])
        else:
            print("-1")