合并K个升序链表 |刷题打卡

305 阅读2分钟

掘金团队号上线,助你 Offer 临门! 点击 查看大厂春招职位

一、题目描述:

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

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

示例 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 = [[]]
输出:[]

题目地址:leetcode-cn.com/problems/me…

二、思路分析:

暴力点的作法是,遍历链表数组,脸部两两依次合并,但是这样的话就会遇到一个问题,当合并到一段时间后,会出现一个链表非常长,另外一个链表却很短,在极端的情况下,可能会需要遍历长链表的所有节点,导致时间复杂度上升。 更好的作法是进行分治的思路,将数组进行分治,直到每组有两个链表的时候,把这两个链表进行归并排序。

三、AC 代码:

public class MergekLists {
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists == null || lists.length == 0) {
            return null;
        }
        return divide(0, lists.length - 1, lists);
    }

    private ListNode divide(int start, int end, ListNode[] listNodes) {
        if (start == end) {
            return listNodes[start];
        }
        ListNode left = divide(start, (start + end) / 2, listNodes);
        ListNode right = divide((start + end) / 2 + 1, end, listNodes);
        return mergeTwoLists(left, right);
    }

    public ListNode mergeTwoLists(ListNode leftList, ListNode rightList) {
        if (leftList == null) return rightList;
        if (rightList == null) return leftList;

        if (leftList.val < rightList.val) {
            leftList.next = mergeTwoLists(leftList.next, rightList);
            return leftList;
        } else {
            rightList.next = mergeTwoLists(leftList, rightList.next);
            return rightList;
        }
    }
}

四、总结:

需要注意的是,我们是直接在原链表上修改的,所以可以使用递归。如果是新建了链表,就要考虑处理结果的问题了。