leetcode--合并K个升序链表

给你一个链表数组,每个链表都已经按升序排列。

请你将所有链表合并到一个升序链表中,返回合并后的链表。


示例 1:

输入:lists = [[1,4,5],[1,3,4],[2,6]]

输出:[1,1,2,3,4,4,5,6]

解释:链表数组如下:

[

  1->4->5,

  1->3->4,

  2->6

]

将它们合并到一个有序链表中得到。

1->1->2->3->4->4->5->6

示例 2:

输入:lists = []

输出:[]

示例 3:

输入:lists = [[]]

输出:[]


提示:

k == lists.length

0 <= k <= 10^4

0 <= lists[i].length <= 500

-10^4 <= lists[i][j] <= 10^4

lists[i] 按 升序 排列

lists[i].length 的总和不超过 10^4

来源:力扣(LeetCode)

链接:https://leetcode-cn.com/problems/merge-k-sorted-lists

著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


代码实现:

class Solution {

    public ListNode mergeKLists(ListNode[] lists) {

        if(lists == null || lists.length == 0) {

            return null;

        }

        List nodes = new ArrayList<>();

        for (ListNode node : lists) {

            while(node != null) {

                nodes.add(node);

                node = node.next;

            }

        }

        if (nodes.isEmpty()){

            return null;

        }

        Collections.sort(nodes, new Comparator() {

            @Override

            public int compare(ListNode o1, ListNode o2) {

                return o1.val - o2.val;

            }

        });

        ListNode head = nodes.get(0);

        for(int i = 0; i < nodes.size(); i++) {

            if (i == nodes.size() - 1) {

                nodes.get(i).next = null;

            } else{

                nodes.get(i).next = nodes.get(i + 1);

            }

        }

        return head;

    }

}

你可能感兴趣的:(leetcode--合并K个升序链表)