题解 | #牛群的合并#

考察链表合并。直接采用归并排序算法应用到上面就可以了。

完整的Java代码如下所示

import java.util.*;

/*
 * public class ListNode {
 *   int val;
 *   ListNode next = null;
 *   public ListNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param lists ListNode类一维数组
     * @return ListNode类
     */
    public ListNode mergeKLists (ListNode[] lists) {
        // write code here
        if (lists == null || lists.length == 0) return null;
        PriorityQueue<ListNode>  pq = new PriorityQueue<>(lists.length,
        new Comparator<ListNode>() {
            public int compare(ListNode p1, ListNode p2) {
                return p1.val - p2.val;
            }
        });
        for (ListNode list : lists) {
            if (list != null)
                pq.add(list);
        }
        ListNode dummyNode = new ListNode(-1), p = dummyNode;
        while (!pq.isEmpty()) {
            ListNode temp = pq.poll();
            if (temp.next != null)
                pq.add(temp.next);
            p.next = temp;
            p = p.next;
        }
        return dummyNode.next;
    }
}