[baekjoon] 1966번 프린터 큐

출처

---

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

1966번: 프린터 큐

 

 

문제

---

프린터 기기는인쇄하고자 하는 문서를 인쇄 명령을 받은 ‘순서대로’, 즉 먼저 요청된 것을 먼저 인쇄한다.

여러 개의 문서가 쌓인다면 Queue 자료구조에 쌓여서 FIFO - First In First Out - 에 따라 인쇄가 되게 된다.

하지만 상근이는 새로운 프린터기 내부 SW를 개발하였는데, 이 프린터기는 다음 조건에 따라 인쇄 하게 된다.

 

    1. 현재 Queue의 가장 앞에 있는 문서의 ‘중요도’를 확인한다.

    2. 나머지 문서들 중 현재 문서보다 중요도가 높은 문서가 하나라도 있다면,

       이 문서를 인쇄하지 않고 Queue의 가장 뒤에 재배치 한다. 그렇지 않다면 바로 인쇄를 한다.

 

예를 들어 Queue에 4개의 문서(A B C D)가 있고, 중요도가 2 1 4 3 라면 C를 인쇄하고,

다음으로 D를 인쇄하고 A, B를 인쇄하게 된다.

여러분이 할 일은, 현재 Queue에 있는 문서의 수와 중요도가 주어졌을 때,

어떤 한 문서가 몇 번째로 인쇄되는지 알아내는 것이다. 예

를 들어 위의 예에서 C문서는 1번째로, A문서는 3번째로 인쇄되게 된다.

 

 

입력

---

첫 줄에 테스트케이스의 수가 주어진다. 각 테스트케이스는 두 줄로 이루어져 있다.

테스트케이스의 첫 번째 줄에는 문서의 개수 N(1 ≤ N ≤ 100)과, 몇 번째로 인쇄되었는지 궁금한 문서가

현재 Queue에서 몇 번째에 놓여 있는지를 나타내는 정수 M(0 ≤ M < N)이 주어진다.

이때 맨 왼쪽은 0번째라고 하자. 두 번째 줄에는 N개 문서의 중요도가 차례대로 주어진다.

중요도는 1 이상 9 이하의 정수이고, 중요도가 같은 문서가 여러 개 있을 수도 있다.

 

 

출력

---

각 테스트 케이스에 대해 문서가 몇 번째로 인쇄되는지 출력한다.

 

 

풀이 및 코드

---

주어진 문서 정보를 enumerate로 for문을 돌면서

(문서의 순서(index), 중요도) 쌍으로 묶어 큐에 담아줬다.

 

이후에는 가장 높은 중요도를 찾아내야 하므로 주어진 문서 정보를 정렬하여 탐색하기 쉽게 구현했다.

나머지 작업은 조건에 맞게 while문 안에서 수행

 

from collections import deque

# 테스트 케이스 t
t = int(input())
for _ in range(t):
    # 문서의 개수 n, 알고자 하는 문서 번호 m
    n, m = map(int, input().split())
    l = list(map(int, input().split()))

    q = deque([])
    for idx, value in enumerate(l):
        # 큐에 문서 순서(idx)와 중요도를 쌍으로 묶어 삽입
        q.append((idx, value))

    # 문서들을 정렬
    l.sort()

    count = 0
    while True:
        # 큐에서 pop
        idx, value = q.popleft()
        # idx가 찾고자하는 m이면서 중요도도 가장 높으면
        # count를 하나 올리고 break
        if idx == m and value == l[-1]:
            count += 1
            break
        # idx가 m은 아니지만 중요도가 가장 높은 문서라면
        # count 후 l에서도 마찬가지로 pop
        elif value == l[-1]:
            count += 1
            l.pop()
        # 나머지 케이스들은 대기 후순위로 배치
        else: q.append((idx,value))
        
    print(count)