1966번: 프린터 큐
여러분도 알다시피 여러분의 프린터 기기는 여러분이 인쇄하고자 하는 문서를 인쇄 명령을 받은 ‘순서대로’, 즉 먼저 요청된 것을 먼저 인쇄한다. 여러 개의 문서가 쌓인다면 Queue 자료구조에
www.acmicpc.net
아이디어: #우선순위큐
파이썬에서 우선순의 큐를 구현하는 방법
# 우선순위가 높은 큐가 먼저 나온다.
import queue
q = queue.PriorityQueue()
q.put((10, "a"))
q.put((5, "b"))
q.put((15, "c"))
q.qsize() # 3
q.get() # (5, 'b')
heapq로 구현하는 방법도 있다.
import heapq
h = []
heapq.heappush(h, (3, "Go to home"))
heapq.heappush(h, (10, "Do not study"))
heapq.heappush(h, (1, "Enjoy!"))
heapq.heappush(h, (4, "Eat!"))
heapq.heappush(h, (7, "Pray!"))
print(h)
#[(1, 'Enjoy!'), (4, 'Eat!'), (3, 'Go to home'), (10, 'Do not study'), (7, 'Pray!')]
근데.. 우선순위 큐를 사용한다고 쉽게 생각했는데, 문제대로 출력이 나오도록 구현하기가 어려웠다.
그래서 구글링함^^
배열(?)이 두 개가 필요하다. 일단 중요도만 우선순위 큐에 넣어서 정렬하고, -> 그냥 배열을 써도 된다. 처음에 오름차순으로 정렬을 한 번 하고, 하나씩 출력될때마다 pop을 하는 방식으로 구현했다. 접근은 p[-1] 이렇게.
중요도와 인덱스를 큐나 덱(덱이 나을 것 같다.)에 넣는다. 그리고 덱을 돌면서 우선순위 큐의 값과 중요도가 같은 경우에는 제거하고, 아닌 경우에는 빼서 다시 뒤에 넣는 식으로..
구하고자 하는 인덱스와 동일한 경우에 그 순위를 출력한다. 순위는 돌면서 계속 카운트 한다.
코드:
import sys
from collections import deque
T = int(sys.stdin.readline())
for _ in range(T):
n, m = map(int, sys.stdin.readline().split())
arr = list(map(int, sys.stdin.readline().split()))
q = []
d = deque()
for i in range(n):
q.append(arr[i])
d.append([i, arr[i]])
q.sort()
cnt=1
while (True):
temp = d.popleft()
if (temp[1]!=q[-1]):
d.append(temp)
else:
if (temp[0]==m):
print(cnt)
break
else:
cnt+=1
q.pop()
오늘 다시 풀어봤다. 다행히..ㅋㅋ 한번에 풀 수 있었다. 함수를 sort 할 때 arr.sort()를 사용해놓고 다른 변수에 넣으려고 해서 에러가 한 번 나긴했지만 sorted 함수로 바꾸어서 해결했다.
import sys
from collections import deque
input = sys.stdin.readline
T = int(input())
for _ in range (T):
N, M = map(int, input().split())
arr = list(map(int, input().split()))
rank = sorted(arr, reverse=True)
q = deque()
ans = [0]*N
for idx in range(len(arr)):
q.append([arr[idx], idx])
cnt = 0
while (q):
if (q[0][0] == rank[cnt]):
ans[q[0][1]] = cnt+1
cnt += 1
q.popleft()
else:
q.append(q.popleft())
print(ans[M])
전체적인 코드는 이전과 비슷하다.
'알고리즘💻 > 스택&큐&덱' 카테고리의 다른 글
BOJ 2504번: 괄호의 값 (0) | 2021.02.01 |
---|---|
BOJ 1021번: 회전하는 큐 (0) | 2021.02.01 |
BOJ 1158번: 요세푸스 문제 / 파이썬 Queue (0) | 2021.01.31 |
BOJ 4949번: 균형잡힌 세상 (0) | 2021.01.31 |
BOJ 1935번: 후위 표기식2 (0) | 2021.01.31 |