[Python] 백준 18258 큐 2 & 시간 초과 이슈

문제

정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램

 

<명령>

• push X: 정수 X를 큐에 넣는 연산
• pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력
• size: 큐에 들어있는 정수의 개수를 출력
• empty: 큐가 비어있으면 1, 아니면 0을 출력
• front: 큐의 가장 앞에 있는 정수를 출력. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력
• back: 큐의 가장 뒤에 있는 정수를 출력. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력

 

 

입력

• 첫째 줄: 명령의 수 N (1 ≤ N ≤ 2,000,000)
• 둘째 줄부터 N개의 줄 : 한 명령 ( 1 ≤ 주어지는 정수 ≤ 100, 000 ) 

 

출력

명령을 수행한 결과

 

 

 

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

18258번: 큐 2

---

★Keypoint : 큐

말 그대로 큐를 사용해서 명령어마다 조건문으로 문제에서 원하는 처리를 해 주면 된다.

 

 

<최종 CODE>

from collections import deque
import sys
queue = deque()
n = int(sys.stdin.readline().rstrip())
for _ in range(n):
    user_input = sys.stdin.readline().split()
    command = user_input[0]
    if command == "push":
        queue.append(int(user_input[1]))
    elif command == "pop":
        print(queue.popleft() if queue else -1)
    elif command == "size":
        print(len(queue))
    elif command == "empty":
        print(1 if not queue else 0)
    elif command == "front":
        print(queue[0] if queue else -1)
    elif command == "back":
        print(queue[-1] if queue else -1)

 

 

 

<CODE 설명>

 

처음에는 시간 초과가 떴는데, 그 이유는 빠른 입력 방법 중 하나인 sys.stdin.readline()을 사용하면 되는 거였다.

코드 작성할 때 습관적으로 input() 을 사용하는데 대량의 입력을 처리하는 경우 상대적으로 느린 특성 때문에 시간 초과가 된 것이다.. 

 

 

deque() 을 활용하면 popleft() 를 사용할 수 있다!