[이·코·테] Q06. 무지의 먹방 라이브

소스 코드 :

import heapq


def solution(food_times, k):
   if sum(food_times) <= k:
       return -1

   queue = []
   for i in range(len(food_times)):
       heapq.heappush(queue, (food_times[i], i + 1))

   sumValue = 0
   before = 0
   length = len(food_times)

   while sumValue + ((queue[0][0] - before) * length) <= k:
       now = heapq.heappop(queue)[0]
       sumValue += (now - before) * length
       length -= 1
       before = now

   result = sorted(queue, key = lambda x : x[1])
   answer = result[(k - sumValue) % length][1]
   return answer

• 문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/42891
• 정답 풀이는 프로그래머스에서 정답 판정을 받았습니다.