[Silver IV] 큐 - 10845
성능 요약
메모리: 31256 KB, 시간: 52 ms
분류
자료 구조, 큐
문제 설명
정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 여섯 가지이다.
- push X: 정수 X를 큐에 넣는 연산이다.
- pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- size: 큐에 들어있는 정수의 개수를 출력한다.
- empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
- front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
입력
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
출력
출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.
🙋 문제 해석
유명한 자료구조의 큐문제입니다. 문제이기에 풀었지만 리스트로 구현한 큐의 단점은 큐의 맨 앞 원소를 삭제하고자 할 때 모든 원소를 앞으로 옮겨야 하는 단점이 있습니다. 이러한 단점은 큐의 원소가 많아질수록 성능 저하를 가져올 수 있습니다. 이를 보완하기 위해서는 collections 모듈에서 제공하는 deque(덱)을 주로 사용합니다.
🔑 Solution 🔑
from sys import stdin
T = int(stdin.readline())
que=[]
for i in range(T):
A = list(stdin.readline().split())
if A[0] == 'push' :
que.append(A[1])
- 입력받고 push처리는 배열로나누어처리했습니다.
원소 삭제 방법elif A[0] == 'pop' : if len(que)!=0 : print(que[0]) que.pop(0)
- 원소를 삭제할 때 del que[0]를 사용해도 되는데, 인덱스를 이용한 삭제 방법은 오류 생길 가능성이 있습니다 que.pop(0)과 같이 pop 메서드를 사용하는 것이 낫습니다.
else :
print(-1)
elif A[0] == 'empty' :
if len(que)==0 :
print(1)
else :
print(0)
elif A[0] == 'size' :
print(len(que))
elif A[0] == 'front' :
if len(que) == 0 :
print(-1)
else :
print(que[0])
elif A[0] == 'back' :
if len(que) == 0 :
print(-1)
else :
print(que[-1])
🕐 시간 복잡도
- 리스트로 구현한 큐에서는 pop(0)과 같은 메서드를 사용할 때마다 리스트의 원소들을 모두 앞으로 당겨와야 하므로, 시간 복잡도가 O(n)입니다. 따라서, 큐의 길이가 길어지면 성능 저하를 일으킬 수 있습니다.
- 다시말하자면 이러한 경우에는 deque와 같은 다른 자료구조를 사용하는 것이 좋습니다.
'코딩테스트 > 백준' 카테고리의 다른 글
[백준][11866] - 요세푸스 문제 python 파이썬 (0) | 2023.04.19 |
---|---|
[백준][1010] - 다리놓기 python (0) | 2023.04.18 |
[백준][1427][Silver 5] - 소트인사이드 python (0) | 2023.04.14 |
[백준][2839][Silver 4] - 설탕배달 python (0) | 2023.04.13 |
[백준][10814][Silver 5] - 나이순 정렬 python (0) | 2023.04.13 |