【Python刷题】合并两个有序列表

豆豆LemonET2024-03-09 13:44

问题描述

将两个升序链表合并为一个新的升序链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。

示例 1：

c 复制代码

输入：l1 = [1,2,4], l2 = [1,3,4]
输出：[1,1,2,3,4,4]

示例 2：

python 复制代码

输入：l1 = [], l2 = []
输出：[]

示例 3：

python 复制代码

输入：l1 = [], l2 = [0]
输出：[0]

解决办法

具体排序结果如上图所示，

首先如果只有一条有序链表，则无序排序

python 复制代码

if list1 is None:
	return list2
if list2 is None:
   	return list1

比较两个链表中数值的大小，如果链表1中的数小于链表2中对应的数，则使链表1中的这个数排在前面，这个数的next指针指向链表1中的下一个数再和链表2中的数值进行比较后的较小的值。
如果链表2中的数小于链表1中的数也是一样的
使用递归方法来完成合并

代码示例

python 复制代码

class Solution:
    def mergeTwoLists(self, list1: Optional[ListNode], list2: Optional[ListNode]) -> Optional[ListNode]:
        if list1 is None:
            return list2
        if list2 is None:
            return list1
        
        if list1.val <= list2.val:
            list1.next = self.mergeTwoLists(list1.next,list2)
            return list1
        else:
            list2.next = self.mergeTwoLists(list1,list2.next)
            return list2

运行结果：