10845 큐

10845번: 큐 (acmicpc.net)

10845번: 큐

import sys
queue = list()
N = int(input())
for i in range(N):
    str = sys.stdin.readline()
    if (str[:4] == 'push'):
        queue.append(int(str[5:]))
    elif (str[:3] == "pop"):
        if (len(queue) != 0):
            print(queue[0])
            queue.pop(0)
        else:
            print(-1)
    elif (str[:5] == "front"):
        print(queue[0] if len(queue) != 0 else -1)
    elif (str[:4] == "back"):
        print(queue[len(queue) - 1] if len(queue) != 0 else -1)
    elif (str[:4] == "size"):
        print(len(queue))
    elif (str[:5] == "empty"):
        print(1 if len(queue) == 0 else 0)

구현은 쉬운데 시간초과 떠서 빡치는 문제.

queue관련 여러 함수들을 만들어보는 건데, 파이썬으로 해서 그런지 간단했다.

단지 시간초과를 위해서 input()대신 sys.stdin.readline()을 써야 한다는 것을 배웠다.

그리고 pop을 앞에서 빼고싶을때는 pop(0)이라고 써야 한다.

또한, 코드의 길이를 줄이기 위해서는 print(1 if ~~ else 0) 이런 식으로 쓸 수 있다.