무지의 먹방 라이브

import java.util.*;

class Solution {
    
    // 음식 정보를 담는 클래스 (먹는 시간, 인덱스)
    public class Food {
        public int time;  // 해당 음식을 먹는 데 걸리는 시간
        public int idx;   // 음식의 원래 인덱스
        
        public Food(int time, int idx){
            this.time = time;
            this.idx = idx;
        }
    }
    
    public int solution(int[] food_times, long k) {
        long len = food_times.length; // 남은 음식 개수
       
        // 먹는 시간이 짧은 순으로 정렬되는 우선순위 큐 생성
        PriorityQueue<Food> pq = new PriorityQueue<>((o1, o2) -> {
            if (o1.time == o2.time)
                return o1.idx - o2.idx; // 시간이 같으면 인덱스 순으로
            return o1.time - o2.time;   // 시간이 짧은 음식이 먼저
        });
        
        // 모든 음식을 큐에 넣기
        for (int i = 0; i < len; i++) {
            pq.add(new Food(food_times[i], i));
        }
        
        long t = 0; // 지금까지 소비된 '기준 시간'

        // 큐가 빌 때까지 반복
        while (!pq.isEmpty()) {
            Food now = pq.peek();        // 가장 적게 남은 음식
            long cur = now.time - t;     // 이전에 제거한 음식 기준으로 남은 시간 차이
            
            // cur * len : 남은 음식 전부를 한 바퀴 돌며 cur초씩 먹는 데 걸리는 총 시간
            if (cur * len <= k) {        
                // 아직 k 안에 다 먹을 수 있다면, 한 사이클 전부 먹기
                t = now.time;            // 이번 음식 시간만큼 기준 시간 갱신
                k -= cur * len;          // 그만큼의 시간 차감
                len--;                   // 한 음식 제거되므로 남은 음식 수 감소
                pq.poll();               // 실제로 음식 제거
                continue;
            }
            
            // 한 바퀴를 다 돌 수 없으면 현재 음식이 먹는 도중에 네트워크가 끊김
            break;
        }
        
        // 모든 음식을 다 먹은 경우 (-1 반환)
        if (pq.size() == 0)
            return -1;
        
        // 아직 남은 음식들을 배열로 옮기기
        Food[] remain = new Food[pq.size()];
        for (int i = 0; i < remain.length; i++) {
            remain[i] = pq.poll();
        }
        
        // 남은 음식들을 원래 인덱스 순서대로 정렬
        Arrays.sort(remain, new Comparator<Food>() {
            @Override
            public int compare(Food o1, Food o2) {
                return o1.idx - o2.idx;
            }
        });
        
        // k초 이후에 먹고 있을 음식의 위치 계산
        Long r = k % remain.length;   // 남은 음식 중 몇 번째인지
        
        // 인덱스 반환
        return remain[r.intValue()].idx + 1;
    }
}