BOJ 1966번: 프린터 큐

www.acmicpc.net/problem/1966

1966번: 프린터 큐

 

아이디어: #우선순위큐

파이썬에서 우선순의 큐를 구현하는 방법

# 우선순위가 높은 큐가 먼저 나온다.
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])

전체적인 코드는 이전과 비슷하다.