백준 1966번. 프린터 큐 (Python / 파이썬)

문제

여러분도 알다시피 여러분의 프린터 기기는 여러분이 인쇄하고자 하는 문서를 인쇄 명령을 받은 ‘순서대로’, 즉 먼저 요청된 것을 먼저 인쇄한다. 여러 개의 문서가 쌓인다면 Queue 자료구조에 쌓여서 FIFO - First In First Out - 에 따라 인쇄가 되게 된다. 하지만 상근이는 새로운 프린터기 내부 소프트웨어를 개발하였는데, 이 프린터기는 다음과 같은 조건에 따라 인쇄를 하게 된다.
 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 이하의 정수이고, 중요도가 같은 문서가 여러 개 있을 수도 있다.

출력

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

---

문제 풀이

프린터 큐 문제의 핵심은 '중요도가 가장 큰 것부터 인쇄' 하는 것입니다.

따라서 중요도를 Queue에 삽입했다면

Queue에 저장된 원소중 가장 큰 원소가 삭제되도록 해야합니다.

 

따라서, Queue의 가장 큰 원소를 저장해준 뒤

Queue의 가장 왼쪽에서 원소 하나를 삭제해 저장해줍니다.

Queue에서 원소 하나가 삭제되면 전체적으로 모든 원소들이 왼쪽으로 이동합니다.

즉, index가 1씩 작아집니다. 따라서 m에서 -1을 빼주어 index를 변경해줍니다.

 

만약 Queue에서 삭제된 값이 max 값이라면 하나가 빠진것이므로

count를 증가시켜 순서를 세줍니다.

이때, 만약 m값이 0보다 작다면 내가 궁금한 원소가 삭제되었다는 뜻입니다.

따라서 count 값을 출력하고 종료시켜줍니다.

 

만약 Queue에서 삭제된 값이 max 값이 아니라면

해당 원소를 다시 Queue의 맨 오른쪽에 삽입해줍니다.

이때, 만약 m값이 0보다 작다면 내가 궁금한 원소가 맨 왼쪽에서 삭제되어

맨 오른쪽에 삽입되었다는 뜻이므로

Queue의 길이에서 1을 뺀 index로 m을 갱신시켜줍니다.

---

My Code

from collections import deque

for tc in range(int(input())): # 테스트케이스 수만큼
    n, m = map(int,input().split()) # 문서 개수, 궁금한 문서가 몇 번째에 놓여있는지

    data = list(map(int,input().split())) # 중요도

    q = deque(data) # 큐에 중요도 삽입

    count = 0 # 몇 번째로 인쇄되는지 count할 변수
    while q:
        max_value = max(q) # 큐의 원소들 중 가장 큰 값
        now = q.popleft()  # 큐에서 삭제된 가장 왼쪽 원소
        m -= 1             # 큐에서 원소가 삭제되면 index가 앞으로 밀림
        
        if now == max_value: # 만약 삭제된 값이 max 값이라면
            count += 1       # 인쇄 순서 증가시켜줌
            if m < 0:        # 만약 index가 0보다 작다면 (궁금한 원소라면)
                print(count) # 순서 출력
                break        

        if now != max_value: # 만약 삭제된 값이 max 값이 아니라면
            q.append(now)    # 큐의 맨 오른쪽에 다시 삽입
            if m < 0:          # 만약 index가 0보다 작다면
                m = len(q) - 1 # 큐의 길이에서 1을 뺀 수로 index 지정