23. Merge k Sorted Lists

public class Solution {
    private static final Comparator<ListNode> listComparator = new Comparator<ListNode>() {
        @Override
        public int compare(ListNode x, ListNode y) {
            return x.val - y.val;
        }
    };
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists.length == 0) return null;
        Queue<ListNode> q = new PriorityQueue<ListNode>(lists.length, listComparator);

        for (ListNode node : lists) {
            if (node != null) {
                q.add(node);
            }
        }

        ListNode dummy = new ListNode(0);
        ListNode prev = dummy;
        while (!q.isEmpty()) {
            ListNode minNode = q.poll();
            prev.next = minNode;
            prev = prev.next;

            if (minNode.next != null) {
                q.add(minNode.next);
            }
        }

        return dummy.next;
    }
}

results matching ""

    No results matching ""