백준 1966. 프린터

문제 설명

프린터는 요청된 것은 먼저 인쇄한다. =

상근이는 새로운 프린터기 내부 소프트웨어를 개발했는데, 이 프린터기는 다음과 같은 조건에 따라 인쇄된다.

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

2. 나머지 문서들 중 현재 문서보다 중요도가 높은 문서가 하나라도 있으면, 이 문서를 인쇄하지 않고 Queue의 가장 뒤에 재배치 한다. 그렇지 않다면 바로 인쇄한다.

 

입력

첫 줄에 테스트케이스의 수가 주어지고

각 테스트케이스는 두 줄로 이루어져 있다.

첫 줄에는 문서의 개수 N과 몇 번째로 인쇄되었는지 궁금한 문서가 Queue에 몇 번째로 놓여 있는지 나타내는 정수 M이 주어진다.

두 번째 줄에는 N개의 문서의 중요도가 차례대로 주어진다. 중요도는 1 이상 9 이하의 정수이고 중요도가 같은 문서가 여러 개 있을 수 있다.

 

출력

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

 

코드

T = int(input())

for i in range(T):
    n, m = map(int, input().split())
    imp = input().split()
    cnt = 0
    impt = []
    for idx, j in enumerate(imp):  # eunumerate는 항상 (idx, 값) 순으로 반환됨
        impt.append((j, idx))
    while impt:
        flag = 0
        for i in range(1, len(impt)):
            if impt[i][0] > impt[0][0]:
                impt.append((impt[0][0], impt[0][1]))  # pop(0) 인자가 있는 pop  /  waiting[0][1] == max(waiting, key=lambda x:x[1])[1]:
                del impt[0]
                flag += 1
                break
        if flag == 1:
            continue
        cnt += 1
        if impt[0][1] == m:
            print(cnt)
        del impt[0]

 

입력으로 주어지는 중요도를 인덱스와 중요도 값을 하나의 튜플로 묶어서 impt 리스트로 관리한다.

enumerate함수는 idx, 값 순서로 값을 반환한다.