23. 合并 K 个升序链表

23. 合并 K 个升序链表

题目描述

给你一个链表数组，每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中，返回合并后的链表。

题解1

思路：使用小根堆，把链表节点放入一个最小堆，每次获得 k 个节点中的最小节点，然后链接在一起。

public ListNode mergeKLists2(ListNode[] lists) {
    if (lists.length == 0) return null;
    // 涉及新链表-虚拟头节点
    ListNode dummy = new ListNode(-1);
    ListNode p = dummy;
    // 优先队列，创建小根堆，堆内有 lists.length 个元素，并指定排序方式
    PriorityQueue<ListNode> pq = new PriorityQueue<>(
            lists.length, (a, b) -> (a.val - b.val)
    );
    // 将 k 个链表的头结点加入最小堆
    for (ListNode head : lists) {
        // 为空入堆
        if (head != null) {
            pq.add(head);
        }
    }
    while (!pq.isEmpty()) {
        // 获得最小节点（堆顶）
        ListNode node = pq.poll();
        p.next = node;
        // node.next非空，入堆，继续构造小根堆
        if (node.next != null) {
            pq.add(node.next);
        }
        // p 指针不断前进
        p = p.next;
    }
    return dummy.next;
}

题解2

思路：每次合并两条链表，合并成为新的链表，新链表再去和另一条合并，直至k条链表全部合并。

public ListNode mergeKLists2(ListNode[] lists) {
    ListNode ans = null;
    for (int i = 0; i < lists.length; i++) {
        ans = mergeTwoLists(ans, lists[i]);
    }
    return ans;
}


public ListNode mergeTwoLists(ListNode list1, ListNode list2) {
    // 虚拟头节点-涉及到新链表
    ListNode dummy = new ListNode(-1);
    ListNode p = dummy;
    ListNode p1 = list1;
    ListNode p2 = list2;
    while (p1 != null && p2 != null) {
        // 比较 p1 和 p2 两个指针
        // 将值较小的的节点接到 p 指针（尾指针）
        if (p1.val <= p2.val) {
            p.next = p1;
            p1 = p1.next;
        } else {
            p.next = p2;
            p2 = p2.next;
        }
        // p 指针不断前进，相当于尾指针
        p = p.next;
    }
    // 若p1或p2非空，接在p后面
    if (p1 != null) {
        p.next = p1;
    }
    if (p2 != null) {
        p.next = p2;
    }
    // 返回新链表（头节点的下一个）
    return dummy.next;
}