(leetcode)力扣100 34合并K个升序链表(排序,分治合并,优先队列)

题目

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

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

数据范围

k == lists.length

0 <= k <= 104

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

-104<= lists[i][j] <= 104

lists[i] 按 升序 排列

lists[i].length 的总和不超过 104

测试用例

示例1

java 复制代码
输入: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

java 复制代码
输入:lists = []
输出:[]

示例3

java 复制代码
输入:lists = [[]]
输出:[]

题解1(博主思路,把这些链表链接起来,然后用上一套题的方法排序,时间nlogn,空间o1)

java 复制代码
/**
 * 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) {
        int length = lists.length;
        // 边界检查:如果是空数组,或者只有一个空链表,直接返回 null
        if (length == 0 || (lists[0] == null && length == 1)) {
            return null;
        }

        ListNode head = null;
        boolean flag = true; // 用于标记是否找到了第一个非空的链表头

        // --- 阶段一:暴力拼接 ---
        // 遍历整个链表数组,将所有非空链表首尾相连
        for (int i = 0; i < length; i++) {
            if (lists[i] != null) {
                // 如果是第一个遇到的非空链表,记录它为最终大链表的头节点
                if (flag) {
                    head = lists[i];
                    flag = false;
                }
                
                // 找到当前链表 lists[i] 的尾节点 (tail)
                ListNode tnode = lists[i];
                while (tnode.next != null) {
                    tnode = tnode.next;
                }

                // 寻找数组中下一个非空的链表
                int pos = i + 1;
                while (pos < length && lists[pos] == null)
                    pos = pos + 1;

                // 如果找到了下一个非空链表,将当前尾节点指向它
                // 实现:List_A_Tail -> List_B_Head
                if (pos < length) {
                    tnode.next = lists[pos];
                    // 更新索引 i,跳过中间那些 null,直接跳到刚刚接上的那个链表的前一个位置
                    // (循环体会执行 i++,所以这里设为 pos-1)
                    i = pos - 1; 
                }
            } else {
                continue;
            }
        }

        // --- 阶段二:全局排序 ---
        // 将拼接好的巨型链表扔进去进行排序
        ListNode res = sortList(head);
        return res;
    }

    // 这是一个标准的【链表归并排序】实现(迭代版/自底向上)
    // 时间复杂度 O(N log N),空间复杂度 O(1)
    public static ListNode sortList(ListNode head) {
        if (head == null) {
            return head;
        }

        // 1. 计算链表总长度
        int length = 0;
        ListNode tnode = head;
        while (tnode != null) {
            length++;
            tnode = tnode.next;
        }

        // 引入哑节点,方便操作头部的变化
        ListNode dummyHead = new ListNode(0, head);

        // 2. 自底向上归并
        // tlen 是子链表的长度,从 1 开始,每次翻倍:1 -> 2 -> 4 -> 8 ...
        for (int tlen = 1; tlen < length; tlen *= 2) {
            ListNode cur = dummyHead.next; // cur 用来遍历链表
            ListNode pre = dummyHead

题解2(顺序合并,时间k²n,空间O1)

java 复制代码
class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        ListNode ans = null;
        for (int i = 0; i < lists.length; ++i) {
            ans = mergeTwoLists(ans, lists[i]);
        }
        return ans;
    }

    public ListNode mergeTwoLists(ListNode a, ListNode b) {
        if (a == null || b == null) {
            return a != null ? a : b;
        }
        ListNode head = new ListNode(0);
        ListNode tail = head, aPtr = a, bPtr = b;
        while (aPtr != null && bPtr != null) {
            if (aPtr.val < bPtr.val) {
                tail.next = aPtr;
                aPtr = aPtr.next;
            } else {
                tail.next = bPtr;
                bPtr = bPtr.next;
            }
            tail = tail.next;
        }
        tail.next = (aPtr != null ? aPtr : bPtr);
        return head.next;
    }
}

题解3(分治合并,时间Oknlogk,空间Ologk)

java 复制代码
/**
 * 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) {
        // 调用分治合并函数,范围是从第一个链表(0)到最后一个链表(length - 1)
        return merge(lists, 0, lists.length - 1);
    }

    /**
     * 核心分治方法
     * 作用:将 lists[l] 到 lists[r] 这一段的链表合并成一个有序链表
     */
    public ListNode merge(ListNode[] lists, int l, int r) {
        // 1. 递归基准条件(Base Case):
        // 当左边界等于右边界时,说明只剩下一个链表了,直接返回它,不需要合并。
        if (l == r) {
            return lists[l];
        }
        
        // 边界处理:防止数组为空的情况 (l > r)
        if (l > r) {
            return null;
        }

        // 2. 「分」(Divide):
        // 找出中间位置。 (l + r) >> 1 等同于 (l + r) / 2,位运算效率更高
        int mid = (l + r) >> 1;

        // 3. 「治」(Conquer):
        // 递归调用:
        // - merge(lists, l, mid): 负责合并左半部分,返回左边合并好的大链表
        // - merge(lists, mid + 1, r): 负责合并右半部分,返回右边合并好的大链表
        // 最后调用 mergeTwoLists 将左右两个大链表合并成最终结果
        return mergeTwoLists(merge(lists, l, mid), merge(lists, mid + 1, r));
    }

    /**
     * 双链表合并方法
     * 作用:合并两个有序链表 a 和 b
     */
    public ListNode mergeTwoLists(ListNode a, ListNode b) {
        // 判空处理:如果其中一个为空,直接返回另一个
        if (a == null || b == null) {
            return a != null ? a : b;
        }

        // 创建哑节点(dummy head),作为结果链表的头部占位符
        ListNode head = new ListNode(0);
        
        // tail 用于指向当前结果链表的尾部
        // aPtr 和 bPtr 作为游标,分别在 a 和 b 链表上移动
        ListNode tail = head, aPtr = a, bPtr = b;

        // 当两个链表都还有节点时,进行比较
        while (aPtr != null && bPtr != null) {
            // 谁小选谁,并将对应的指针后移
            if (aPtr.val < bPtr.val) {
                tail.next = aPtr;
                aPtr = aPtr.next;
            } else {
                tail.next = bPtr;
                bPtr = bPtr.next;
            }
            // 尾指针永远指向最新的节点
            tail = tail.next;
        }

        // 收尾工作:
        // 循环结束后,肯定有一个链表先走完。
        // 将 tail 直接指向剩下的那个链表(因为剩下的肯定是有序的,直接接上即可)
        tail.next = (aPtr != null ? aPtr : bPtr);

        // 返回哑节点的下一个节点,即真正的头节点
        return head.next;
    }
}

题解4(优先队列,时间Oknlogk,空间Ok)

java 复制代码
class Solution {
    // 定义一个包装类 Status,用于放入优先队列中
    // 实现了 Comparable 接口,以便优先队列能自动根据 val 值进行排序
    class Status implements Comparable<Status> {
        int val;
        ListNode ptr;

        Status(int val, ListNode ptr) {
            this.val = val;
            this.ptr = ptr;
        }

        // 定义比较规则:从小到大排序(最小堆)
        public int compareTo(Status status2) {
            return this.val - status2.val;
        }
    }

    // 创建一个小顶堆(PriorityQueue),队列头部永远是 val 最小的节点
    PriorityQueue<Status> queue = new PriorityQueue<Status>();

    public ListNode mergeKLists(ListNode[] lists) {
        // 1. 初始化阶段:
        // 遍历所有链表的头节点。只要链表非空,就把它放入队列。
        // 这样队列里就包含了所有链表当前的"领头羊"。
        for (ListNode node: lists) {
            if (node != null) {
                queue.offer(new Status(node.val, node));
            }
        }

        // 创建一个哑节点(dummy head),用于方便地构建结果链表
        ListNode head = new ListNode(0);
        ListNode tail = head;

        // 2. 循环合并阶段:
        // 只要队列不为空,就说明还有节点没处理完
        while (!queue.isEmpty()) {
            // A. 从队列中取出当前最小的节点 f (poll操作)
            Status f = queue.poll();
            
            // B. 将这个最小节点接到结果链表的末尾
            tail.next = f.ptr;
            tail = tail.next;

            // C. 关键步骤(补充弹药):
            // 如果刚刚取出的那个节点所在的链表后面还有节点,
            // 就把那个链表的"下一个节点"入队。
            // 这样能保证队列里始终维护着 k 个链表当前的头部候选人。
            if (f.ptr.next != null) {
                queue.offer(new Status(f.ptr.next.val, f.ptr.next));
            }
        }
        
        // 返回哑节点的下一个节点,即真正的结果链表头
        return head.next;
    }
}

思路

这道题虽然整体看起来代码量大,其实思路不难,如果100题的链表是一道一道写过来的,这道题都能较为轻松的解决,唯一需要住的是,链表中可能存在空链表,用有些方法的时候,需要对其进行一个判定。

因为博主刚做了排序,就直接干脆用排序做了,当然时间复杂度肯定是不优的,但空间复杂度能做到O1。当然大家还是看其他几个方法比较好,写起来更方便,效果还更优。

追求时间复杂度的话使用优先队列与分治法,这两者都不难,能想起来有这个方法就是成功。

追求空间复杂度就顺序合并,我连备注都不想加,太简单了。

相关推荐
橘颂TA2 小时前
【剑斩OFFER】算法的暴力美学——二进制求和
算法·leetcode·哈希算法·散列表·结构与算法
地平线开发者3 小时前
征程 6 | cgroup sample
算法·自动驾驶
尋有緣4 小时前
力扣1355-活动参与者
大数据·数据库·leetcode·oracle·数据库开发
姓蔡小朋友4 小时前
算法-滑动窗口
算法
君义_noip4 小时前
信息学奥赛一本通 2134:【25CSPS提高组】道路修复 | 洛谷 P14362 [CSP-S 2025] 道路修复
c++·算法·图论·信息学奥赛·csp-s
kaikaile19954 小时前
基于拥挤距离的多目标粒子群优化算法(MO-PSO-CD)详解
数据结构·算法
不忘不弃4 小时前
求两组数的平均值
数据结构·算法
leaves falling4 小时前
迭代实现 斐波那契数列
数据结构·算法
珂朵莉MM5 小时前
全球校园人工智能算法精英大赛-产业命题赛-算法巅峰赛 2025年度画像
java·人工智能·算法·机器人