[백준 문제풀이] 18258 - 제로

백준 문제풀이 18258번

풀이

명령을 split()으로 나눠서 [0]과 [1]부분으로 나누고, [0]부분에 명령어 부분을 받아서 처리한다.

코드

import sys

input = sys.stdin.readline

N = int(input())
Q = []
count = 0
for i in range(N):
    order = input().rstrip()
    split = order.split()
    if split[0] == "push":
        Q.append(int(order.split()[1]))
    elif split[0] == "pop":
        if len(Q) - count == 0:
            print(-1)
        else:
            print(Q[count])
            count += 1

    elif split[0] == "size":
        print(len(Q) - count)
    elif split[0] == "empty":
        if len(Q) - count == 0:
            print(1)
        else:
            print(0)
    elif split[0] == "front":
        if len(Q) - count == 0:
            print(-1)
        else:
            print(Q[count])
    elif split[0] == "back":
        if len(Q) - count == 0:
            print(-1)
        else:
            print(Q[-1])