[프로그래머스-Lv2] 프린터 / Python

코딩테스트 연습 - 프린터

import sys

input = sys.stdin.readline

priorities = list(map(int, input().split()))
location = int(input())


def solution(priorities, location):
    priorities = [(priorities[x], x) for x in range(len(priorities))]
    answer = 0
    while True:
        if priorities[0][0] == max(priorities)[0] and priorities[0][1] == location:
            answer += 1
            break
        if priorities[0][0] < max(priorities)[0]:
            num, idx = priorities.pop(0)
            priorities.append((num, idx))
        else:
            answer += 1
            priorities.pop(0)
    return answer


print(solution(priorities, location))

 

해설

리스트에 priorities의 값과 인덱스로 튜플형식으로 저장해준다.

1) 리스트의 맨 앞 튜플에 저장된 값이 리스트의 최대이면서 리스트의 맨 앞 튜플에 저장된 인덱스가 location과 일치하면 반복문을 중단한다.

2) 리스트의 맨 앞 튜플에 저장된 값이 리스트의 최대보다 작으면 튜플에 저장된 값과 인덱스를 pop시키고 다시 append 시킨다.

3) 리스트의 맨 튜플에 저장된 값이 최대이면 제거한다.