Create PG_무지의먹방라이브.java

Author: kn9012

250303/PG_무지의먹방라이브.java

@@ -0,0 +1,55 @@
+import java.util.*;
+
+class Solution {
+    public class Food implements Comparable<Food> {
+        int index, time;
+
+        Food(int index, int time) {
+            this.index = index;
+            this.time = time;
+        }
+
+        @Override
+        public int compareTo(Food o) {
+            return Integer.compare(this.time, o.time);
+        }
+    }
+
+    public int solution(int[] food_times, long k) {
+        long sum = 0;
+
+        PriorityQueue<Food> queue = new PriorityQueue<>();
+
+        for (int i = 0; i < food_times.length; i++) {
+            queue.add(new Food(i, food_times[i]));
+        }
+
+        int prevTime = 0;
+
+        while (!queue.isEmpty()) {
+            int size = queue.size();
+            Food cur = queue.peek();
+
+            long cost = (long) size * (cur.time - prevTime);
+            if (sum + cost <= k) {
+                sum += cost;
+                prevTime = cur.time;
+                queue.poll();
+            } else {
+                k -= sum;
+                break;
+            }
+        }
+
+        if (queue.isEmpty()) return -1;
+
+        List<Food> list = new ArrayList<>();
+        while (!queue.isEmpty()) {
+            list.add(queue.poll());
+        }
+
+        list.sort(Comparator.comparingInt(o -> o.index));
+
+        return list.get((int) (k % list.size())).index + 1;
+    }
+}