[Python][백준 10845] 큐

📚 큐 문제 풀이

백준 10845번 큐 문제의 파이썬 풀이

📝 문제 설명

💡 풀이

이 문제는 큐의 기본적인 연산들을 구현하는 문제이다. 파이썬의 collections 모듈의 deque를 사용하여 큐를 구현하고, 각각의 명령어에 따라 적절한 동작을 수행하면 된다.

🔍 코드 설명

1. 명령의 수 N을 입력받는다.
2. N번 반복하면서:
   • 명령어를 입력받아 분석한다.
   • 각 명령어에 맞는 동작을 수행한다.
   • 결과를 출력해야 하는 명령의 경우 결과값을 저장한다.
3. 저장된 모든 결과를 한 번에 출력한다.

✨ 참고

• deque는 양방향 큐로, 앞과 뒤에서 모두 효율적으로 요소를 추가/제거할 수 있다.
• append()로 요소를 추가하고 popleft()로 제거한다.
• sys.stdin.readline()을 사용하여 입력 속도를 개선할 수 있다.
• sys.stdout.write()를 사용하여 출력 속도를 개선할 수 있다.

🎯 주의사항

• 큐가 비어있을 때 pop, front, back 명령이 들어올 수 있다.
• 시간 제한이 있으므로 입출력 최적화가 필요하다.

---

📝 코드

import sys
from collections import deque

n = int(sys.stdin.readline().strip())
queue = deque()
results = []

for _ in range(n):
    command = sys.stdin.readline().strip().split()
    
    if command[0] == "push":
        queue.append(int(command[1]))
    elif command[0] == "pop":
        results.append(queue.popleft() if queue else -1)
    elif command[0] == "size":
        results.append(len(queue))
    elif command[0] == "empty":
        results.append(0 if queue else 1)
    elif command[0] == "front":
        results.append(queue[0] if queue else -1)
    elif command[0] == "back":
        results.append(queue[-1] if queue else -1)

sys.stdout.write("\n".join(map(str, results)) + "\n")

Python Coding Test 카테고리 내 다른 글 보러가기