23.合并k个升序序链表- 力扣(LeetCode)

题目:

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

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

示例 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 <= listsi.length <= 500

  • -10^4 <= listsi <= 10^4

  • listsi 按 升序 排列

  • listsi.length 的总和不超过 10^4

思路如下:
方法一:最小堆

合并后的第一个节点 first,一定是某个链表的头节点(因为链表已按升序排列)。

合并后的第二个节点,可能是某个链表的头节点,也可能是 first 的下一个节点。

例如有三个链表 1->2->5, 3->4->6, 4->5->6,找到第一个节点 1 之后,第二个节点不是另一个链表的头节点,而是节点 1 的下一个节点 2。

按照这个过程继续思考,每当我们找到一个节点值最小的节点 x,就把节点 x.next 加入「可能是最小节点」的集合中。

因此,我们需要一个数据结构,它支持:

  • 从数据结构中找到并移除最小节点。

  • 插入节点。

这可以用最小堆实现。初始把所有链表的头节点入堆,然后不断弹出堆中最小节点 x,如果 x.next 不为空就加入堆中。循环直到堆为空。把弹出的节点按顺序拼接起来,就得到了答案。

注意:

**问题:**当多个节点的 val 相同时,Python 尝试比较 ListNode 对象,但未定义比较规则会报错。

**修复:**在元组中增加 index 作为次要排序键,确保即使 val 相同,堆也能正确比较。

方法二:分治(迭代)

结合21题思路,直接自底向上合并链表:

  • 两两合并:把 lists0 和 lists1 合并,合并后的链表保存在 lists0 中;把 lists2 和 lists3 合并,合并后的链表保存在 lists2 中;依此类推。

  • 四四合并:把 lists0 和 lists2 合并(相当于合并前四条链表),合并后的链表保存在 lists0 中;把 lists4 和 lists6 合并,合并后的链表保存在 lists4 中;依此类推。

  • 八八合并:把 lists0 和 lists4 合并(相当于合并前八条链表),合并后的链表保存在 lists0 中;把 lists8 和 lists12 合并,合并后的链表保存在 lists8 中;依此类推。

  • 依此类推,直到所有链表都合并到 lists0 中。最后返回 lists0

题解如下:
方法一:最小堆
python 复制代码
import heapq	# 导入堆模块
class Solution:
    def mergeKLists(self , lists):
        """
                :type:  lists: List[ListNode]
                :rtype: ListNode
        """
        # write code here
        # 初始化堆,存储 (节点值, 唯一索引, 节点) 的元组
        h = []
        index = 0	# 全局唯一索引,确保相同值时堆能正确比较
        for node in lists:
            if node:	# 过滤空链表
                heapq.heappush(h,(node.val, index, node))	# 将节点值、索引、节点存入堆(按值排序)
                index += 1	# 索引递增,保证唯一性
        
        dummy = ListNode(0)
        cur = dummy
               
        while h:
            val,idx,node = heapq.heappop(h)	# 使用 idx 接收弹出索引
            cur.next = node	# 将节点连接到结果链表
            cur = cur.next	# 移动当前指针

            # 若当前节点有后继节点,将其加入堆
            if node.next:
                heapq.heappush(h, (node.next.val, index, node.next))
                index += 1	# 全局索引递增
        return dummy.next	# 返回合并后的链表头
方法二:分治(迭代)
python 复制代码
class Solution:
    def mergeKLists(self, lists):
        """
                :type:  lists: List[Optional[ListNode]]
                :rtype: Optional[ListNode]
        """
        m = len(lists)
        if m == 0:
            return None	# 空列表直接返回 None
        step = 1
        while step < m:	# 分治合并循环
            # 遍历合并相邻的两个链表,间隔为 step*2
            for i in range(0, m - step, step * 2):
                lists[i] = self.mergeTwoLists(lists[i], lists[i + step])
            step *= 2	# 合并范围扩大一倍
        return lists[0]	# 最终合并结果在 lists[0]
    
    # 21. 合并两个有序链表
    def mergeTwoLists(self, list1, list2):
        """
                :type:  list1: Optional[ListNode], list2: Optional[ListNode]
                :rtype: Optional[ListNode]
        """
        cur = dummy = ListNode()  # 用哨兵节点简化代码逻辑
        while list1 and list2:
            if list1.val < list2.val:
                cur.next = list1  # 把 list1 加到新链表中
                list1 = list1.next
            else:  # 注:相等的情况加哪个节点都是可以的
                cur.next = list2  # 把 list2 加到新链表中
                list2 = list2.next
            cur = cur.next
        cur.next = list1 if list1 else list2  # 拼接剩余链表
        return dummy.next
相关推荐
原来是猿12 分钟前
博客系统自动化测试实战总结
python
小江的记录本41 分钟前
【JVM虚拟机】JVM调优:常用JVM参数、调优核心指标、OOM排查、GC日志分析、Arthas工具使用(附《思维导图》+《面试高频考点清单》)
java·jvm·spring boot·后端·python·spring·面试
大数据魔法师43 分钟前
Streamlit(十三)- API 参考文档(六)- 媒体展示组件
python·web
爱写代码的倒霉蛋1 小时前
Hello-Agents的第一个练习-5分钟实现一个智能体(实现详解)
python
金銀銅鐵1 小时前
[Java] 用图形化界面演示 iadd, isub, iconst_<i> 指令的效果
java·后端·python
云泽8081 小时前
笔试算法 - 链表篇(一):移除、反转、合并、回文判断全解析
数据结构·c++·算法·链表
菜菜的顾清寒1 小时前
HOT力扣100(43)二叉树-翻转二叉树
数据结构·算法·leetcode
春日见1 小时前
五分钟入门 强化学习---DQN(Deep Q Net)算法与实现
人工智能·python·深度学习·算法·microsoft·机器学习
SomeOtherTime1 小时前
Geojson相关(AI回答)
java·前端·python
weelinking2 小时前
【产品】11_实现后端接口——数据在背后如何流动
java·人工智能·python·sql·oracle·json·ai编程