코딩 테스트/백준

[Python] 백준 18258 큐 2

위시리 2024. 9. 11. 01:09

 

문제 분석

  • 정수를 저장하는 큐 구현하기
  • 6가지 명령

 

코드 설계

  • 입력 n : 명령 수 (반복 횟수)
  • push 명령만 입력값 x를 같이 입력받아야 한다.
  • 명령마다 출력
  • 두개를 입력받을 수 있으니, 입력 받을 때 list로 받음

 

정답 코드

import sys
input = sys.stdin.readline
from collections import deque

n = int(input())
d = deque()
command = []

for i in range(n):
    command = input().split()
    if command[0] == "push" :
        d.append(command[1])
        # print(d[-1])
    elif command[0] == "pop" :
        if d :
            print(d.popleft())
        else :
            print(-1)
    elif command[0] == "size" :
        print(len(d))
    elif command[0] == "empty" :
        if d :
            print(0)
        else :
            print(1)
    elif command[0] == "front" :
        if d :
            print(d[0])
        else :
            print(-1)
    elif command[0] == "back" :
        if d :
            print(d[-1])
        else :
            print(-1)