[백준BOJ] 10845번 큐.py

 

백준 저지에서 큐를 파이썬을 통해 풀어 보았다. 

 

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

10845번: 큐

10845번 큐.py

tomy9729/Algorithm

 

 

 

10845번 큐

문제

정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 여섯 가지이다.

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

 

설명

사실 큐에 관한 개념을 알고 있으면 바로 풀 수 있는 문제이다. 단 파이썬으로 풀게 되면 시간 초과와 관련하여 몇 가지 문제가 발생한다. 파이썬에서는 list를 사용하여 큐를 구현할 수 있는데 이 경우 파이썬에서 제공하는 deque를 사용하는 게 훨씬 빠르다.

 

또한 추가로 시간 초과를 회피하기 위해 input대신 sys.stdin.readline을 사용한다. 하지만 그냥 사용하게 되면 문제에서 틀렸다고 나온다.  sys.stdin.readline의 경우 마지막에 개행 문자까지 넘겨주면서 명령어가 인식되지 않기 때문이다. 개행 문자를 제거하고 사용하면 해결할 수 있다.

코드

#10845번 큐.py
import sys
input = sys.stdin.readline
from collections import deque
if __name__ == "__main__":
  q = deque()
  n = int(input())
  for i in range(n) : 
    instruntion = input()[:-1]
    if instruntion[:4] == "push" : 
      q.append(int(instruntion[5:]))
    elif instruntion == "pop" : 
      if len(q) == 0 : 
        print(-1)
      else :
        print(q.popleft())
    elif instruntion == "size" : 
      print(len(q))
    elif instruntion == "empty" : 
      if len(q) == 0 :
        print(1)
      else : 
        print(0)
    elif instruntion == "front" : 
      if len(q) == 0 :
        print(-1)
      else :
        print(q[0])
    elif instruntion == "back" : 
      if len(q)==0:
        print(-1)
      else :
        print(q[-1])