백준(baekjoon) 18258번: 큐 2(실버 4, 파이썬 PYTHON) - 큐,덱

문제

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

명령은 총 여섯 가지이다.

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

 

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

18258번: 큐 2

 

import sys
from collections import deque

N=int(sys.stdin.readline())
queue=deque()

for _ in range(N):
  command=sys.stdin.readline().split()
  
  if(command[0]=='push'):
    queue.append(command[1])

  if(command[0]=='pop'):
    if(len(queue)!=0):
      print(queue[0])
      queue.popleft()
    else:
      print(-1)

  if(command[0]=='size'):
    print(len(queue))

  if(command[0]=='empty'):
    if(len(queue)==0):
      print(1)
    else:
      print(0)

  if(command[0]=='front'):
    if(len(queue)!=0):
      print(queue[0])
    else:
      print(-1)

  if(command[0]=='back'):
    if(len(queue)!=0):
      print(queue[len(queue)-1])
    else:
      print(-1)

 

※ 큐, 덱

 

 

● 먼저 주어지는 명령의 수 N을 입력받아 준 후, queue라는 deque을 선언해준다.

 

● for문은 총 N번 돌게 된다. 

 

- 입력 받은 command[0]이 'push' 와 같을 경우:

-> command[1] 에 들어있는 숫자를 queue 에 append 해준다.

 

- 입력 받은 command[0]이 'pop' 과 같을 경우:

-> queue의 길이가 0이면 아니면 맨 앞 원소 queue[0]을 출력한 뒤, 덱의 함수 popleft()를 활용하여 맨 앞 원소를 queue에서 제거한다.

 

- 입력 받은 command[0]이 'size' 와 같을 경우:

-> queue의 길이 len(queue)을 출력해준다.

 

- 입력 받은 command[0]이 'empty' 와 같을 경우:

-> queue의 길이가 0이면 1을 출력해주고 길이가 0이 아니면 0을 출력해준다.

 

- 입력 받은 command[0]이 'front' 과 같을 경우:

-> queue의 길이가 0이 아니면 맨 앞 원소 queue[0] 을 출력해주고 queue의 길이가 0이면 -1을 출력해준다.

 

- 입력 받은 command[0]이 'back'과 같을 경우:

-> queue의 길이가 0이 아니면 맨 끝 원소 queue[len(queue)-1] 를 출력해주고 queue의 길이가 0이면 -1을 출력해준다.

 

 

 

♣ 참고: 덱(deque)

-> https://wikidocs.net/104977

008 앞뒤에서 자료를 넣고 빼려면? ― collections.deque