【NO.14】LeetCode HOT 100—23. 合并 K 个升序链表

23. 合并 K 个升序链表

23. 合并 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

解题

方法一：顺序合并

顺序合并，逐个将数组的中的合并到结果中，时间复杂度为O(k^2 n) 链表的最长长度是 n，k为链表长度，空间复杂度为O（1）

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    // 顺序合并, ans ,将链表一个个往前合并
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists.length == 0) {
            return null;
        }

        ListNode ans = null;
        for (int i = 0; i < lists.length; i++) {
            ans = mergeTwoListNode(ans, lists[i]);
        }
        return ans;
    }
	
	// 合并两个有序链表，返回头节点
    private ListNode mergeTwoListNode(ListNode l1, ListNode l2) {
        if (l1 == null) {
            return l2;
        }
        if (l2 == null) {
            return l1;
        }

        ListNode head = new ListNode(0); 
        ListNode temp = head;
        ListNode tempL1 = l1;
        ListNode tempL2 = l2;
        while (tempL1 != null && tempL2 != null) {
            if (tempL1.val <= tempL2.val) {
                temp.next = tempL1;
                tempL1 = tempL1.next;
            } else {
                temp.next = tempL2;
                tempL2 = tempL2.next;
            }

            temp = temp.next;
        }

        if (tempL1 != null) {
            temp.next = tempL1;
        }
        if (tempL2 != null) {
            temp.next = tempL2;
        }

        return head.next;
    }
}

方法二：分治合并

时间复杂度：考虑递归「向上回升」的过程——第一轮合并 k/2组链表，每一组的时间代价是 O(2n)；第二轮合并 k/4组链表，每一组的时间代价是 O(4n)…所以总的时间代价是 O(kn×logk)，故渐进时间复杂度为 O(kn×log⁡k)

空间复杂度：递归会使用到 O(log⁡k) 空间代价的栈空间。

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    
    public ListNode mergeKLists(ListNode[] lists) {
        // 分治，先两两合并，最后归一
        if (lists.length == 0) {
            return null;
        }

        return merge(lists, 0, lists.length-1);
    }

    private ListNode merge(ListNode[] lists, int left, int right) {
        // 划分到了最后
        if (left == right) {
            return lists[left];
        }
        if (left > right) {
            return null;
        }
        // 中间节点
        int mid = (left + right) >> 1;
        return mergeTwoListNode(merge(lists, left, mid), 
        							merge(lists, mid+1, right));

    }

    // 合并两个升序链表
    private ListNode mergeTwoListNode(ListNode node1, ListNode node2) {
        ListNode head1 = node1;
        ListNode head2 = node2;
        ListNode temp = new ListNode(-1);
        ListNode ans = temp;
        while (head1 != null && head2 != null) {
            if (head1.val < head2.val) {
                temp.next = head1;
                head1 = head1.next;
            } else {
                temp.next = head2;
                head2 = head2.next;
            }
            temp = temp.next;
        }

        if (head1 != null) {
            temp.next = head1;
        }
        if (head2 != null) {
            temp.next = head2;
        }
        return ans.next;
    }


}

方法三：优先队列

这个方法和前两种方法的思路有所不同，我们需要维护当前每个链表没有被合并的元素的最前面一个，k 个链表就最多有 k 个满足这样条件的元素，每次在这些元素里面选取 val属性最小的元素合并到答案中。在选取最小元素的时候，我们可以用优先队列来优化这个过程。

时间复杂度：考虑优先队列中的元素不超过 k 个，那么插入和删除的时间代价为 O(log⁡k)，这里最多有 kn 个点，对于每个点都被插入删除各一次，故总的时间代价即渐进时间复杂度为 O(kn×logk)。

空间复杂度：这里用了优先队列，优先队列中的元素不超过 k 个，故渐进空间复杂度为 O(k)

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists.length == 0) {
            return null;
        }
		// 初始化优先队列
        PriorityQueue<ListNode> queue = new PriorityQueue<ListNode>(
            new Comparator<ListNode>(){
                public int compare(ListNode l1, ListNode l2) {
                    return l1.val - l2.val;
                }
            }
        );
		// 将所有元素放进优先队列
        for (ListNode list : lists) {
            if (list != null) {
                queue.add(list);
            }
        }

        ListNode ans = new ListNode(-1);
        ListNode temp = ans;
        while(!queue.isEmpty()) {
        	// 每次取最小的那个（小顶堆） 
            ListNode node = queue.poll();
            if (node != null) {
                temp.next = node;
                temp = temp.next;
            }
            // 将下一个链表节点放进队列里，重新构建小顶堆
            if (node.next != null) {
                queue.add(node.next);
            }
        }

        return ans.next;
       
    }

   



}