[백준] 18285 : 큐2 (Python/파이썬)

문제 링크

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

문제

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

명령은 총 여섯 가지이다.

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

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 2,000,000)이 주어진다. 
둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 
주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 
문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

예제

입력

15
push 1
push 2
front
back
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
front

출력

1
2
2
0
1
2
-1
0
1
-1
0
3

---

정답 및 풀이

import sys
from collections import deque

input = sys.stdin.readline

n = int(input())
q = deque()
out = []

for _ in range(n):
    s = input().split()

    if s[0] == "push":
        q.append(int(s[1]))
    elif s[0] == "pop":
        out.append(str(q.popleft() if q else -1))
    elif s[0] == "size":
        out.append(str(len(q)))
    elif s[0] == "empty":
        out.append("1" if not q else "0")
    elif s[0] == "front":
        out.append(str(q[0] if q else -1))
    elif s[0] == "back":
        out.append(str(q[-1] if q else -1))

print("\\n".join(out))

1. import sys
   • 빠른 입력을 사용하기 위해 sys 모듈을 불러온다.
2. from collections import deque
   • 큐를 효율적으로 구현하기 위해 deque를 사용한다.
   • 앞뒤 삽입·삭제가 모두 O(1)이다.
3. input = sys.stdin.readline
   • 기본 input() 대신 더 빠른 입력 방식을 사용한다.
4. n = int(input())
   • 명령의 개수 n을 입력받는다.
5. q = deque()
   • 실제 큐 역할을 할 deque를 생성한다.
6. out = []
   • 출력 결과를 저장할 리스트를 만든다.
   • print를 여러 번 호출하는 대신 한 번에 출력하기 위함이다.
7. for _ in range(n):
   • 명령의 개수만큼 반복한다.
8. s = input().split()
   • 한 줄의 명령을 공백 기준으로 나누어 리스트로 저장한다.
9. if s[0] == "push":
   • 명령이 push인 경우를 처리한다.
10. q.append(int(s[1]))
    • 큐의 뒤쪽에 정수를 하나 추가한다.
11. elif s[0] == "pop":
    • 명령이 pop인 경우를 처리한다.
12. out.append(str(q.popleft() if q else -1))
    • 큐가 비어 있지 않으면 맨 앞 원소를 꺼내서 저장한다.
    • 비어 있으면 1을 저장한다.
13. elif s[0] == "size":
    • 명령이 size인 경우를 처리한다.
14. out.append(str(len(q)))
    • 현재 큐에 들어 있는 원소의 개수를 저장한다.
15. elif s[0] == "empty":
    • 명령이 empty인 경우를 처리한다.
16. out.append("1" if not q else "0")
    • 큐가 비어 있으면 1, 아니면 0을 저장한다.
17. elif s[0] == "front":
    • 명령이 front인 경우를 처리한다.
18. out.append(str(q[0] if q else -1))
    • 큐가 비어 있지 않으면 맨 앞 원소를 저장한다.
    • 비어 있으면 1을 저장한다.
19. elif s[0] == "back":
    • 명령이 back인 경우를 처리한다.
20. out.append(str(q[-1] if q else -1))
    • 큐가 비어 있지 않으면 맨 뒤 원소를 저장한다.
    • 비어 있으면 1을 저장한다.
21. print("\\n".join(out))
    • 저장해 둔 출력 결과를 줄바꿈으로 연결해 한 번에 출력한다.

---