[백준] 1966번 프린터 큐 (파이썬)

1966번: 프린터 큐

 

코드

from collections import deque

def print_que(list_num, want, array):
    check_li = [0 for i in range(list_num)]
    check_li[want] = 1
    cnt = 0
    for i in range(list_num):
        while array[0] != max(array):
            x = array.pop(0)
            z = check_li.pop(0)
            array.append(x)
            check_li.append(z)
        cnt += 1
        array.pop(0)
        y = check_li.pop(0)
        if y == 1:
            return cnt

n = int(input())
for i in range(n):
    list_num, want = map(int, input().split())
    array = list(map(int, input().split()))
    a = print_que(list_num, want, array)
    print(a)