[백준 10828] 스택 with Python

문제 링크

www.acmicpc.net/problem/10828

10828번: 스택

풀이

import sys

n = int(input())
stack = []
for i in range(n):
    command = sys.stdin.readline().split()
    if len(command) == 2:
        stack.append(int(command[1]))
    else:
        if command[0] == "pop":
            if stack:
                print(stack.pop())
            else:
                print(-1)
        elif command[0] == "size":
            print(len(stack))
        elif command[0] == "empty":
            if stack:
                print(0)
            else:
                print(1)
        elif command[0] == "top":
            if stack:
                print(stack[-1])
            else:
                print(-1)

입력된 명령어를 split()하여 길이가 2이면 push X로 분류, 그렇지 않으면 명령어의 이름에 따라 분기하여 각각의 명령을 실행하도록 하였다. 처음에 sys.stdin.readline() 대신 input()을 사용하여 시간 초과가 났다. 앞으로는 어떤 문제를 풀든지 반드시 표준입력을 사용해야겠다.