[백준] 10828 스택

📖 문제

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

💻 내 코드

import sys

n=int(sys.stdin.readline())
mlist=[]

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

💡 풀이

• 시간초과를 방지하기 위해 입력을 sys.stdin.readline()로 받음

x=sys.stdin.readline().split()
w=x[0] //split으로 분리된 입력 중 0번째를 가져옴