update 23

fishercoder1534 · fishercoder1534 · commit 14e9ee97d822 · 2024-06-09T17:13:32.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_23.java b/src/main/java/com/fishercoder/solutions/_23.java
@@ -2,13 +2,12 @@
 
 import com.fishercoder.common.classes.ListNode;
 
-import java.util.Comparator;
 import java.util.PriorityQueue;
 
 public class _23 {
     public static class Solution1 {
         public ListNode mergeKLists(ListNode[] lists) {
-            PriorityQueue<ListNode> heap = new PriorityQueue((Comparator<ListNode>) (o1, o2) -> o1.val - o2.val);
+            PriorityQueue<ListNode> heap = new PriorityQueue<>((a, b) -> a.val - b.val);
 
             for (ListNode node : lists) {
                 if (node != null) {
@@ -21,10 +20,10 @@ public ListNode mergeKLists(ListNode[] lists) {
             while (!heap.isEmpty()) {
                 ListNode curr = heap.poll();
                 temp.next = new ListNode(curr.val);
+                temp = temp.next;
                 if (curr.next != null) {
                     heap.offer(curr.next);
                 }
-                temp = temp.next;
             }
             return pre.next;
         }
