백준 1966 : 프린터 큐 with 파이썬

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

1966번: 프린터 큐

 

# 포인트

1. enumerate로 (값과 인덱스)를 묶어주기

2. 최대값인지 확인 : max(iterable, key=lambda x:)

값이 최대값이면 cnt+1 하고 바로 출력하고, q.popleft()

최대값이 아니면 뒤에 다시 넣기 q.append(q.popleft())

3. cnt로 출력순서 갱신

 

import sys
from collections import deque
input=sys.stdin.readline
case=int(input())
for _ in range(case):
    n,m=map(int, input().split())
    nums=list(map(int, input().split()))
    q=deque([(i, idx) for idx, i in enumerate(nums)])

    cnt=0
    while True:
        if q[0][0] == max(q, key=lambda x : x[0])[0]:   #q[0][0]는 값
            cnt+=1
            if q[0][1]==m:                              #q[0][1]는 인덱스
                print(cnt)
                break
            else :
                q.popleft()
        else :
            q.append(q.popleft())