leetCode第21题 合并两个有序链表
将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例 1:
输入:l1 = [1,2,4], l2 = [1,3,4]
输出:[1,1,2,3,4,4]
示例 2:
输入:l1 = [], l2 = []
输出:[]
示例 3:
输入:l1 = [], l2 = [0]
输出:[0]
提示:
两个链表的节点数目范围是 [0, 50]
-100 <= Node.val <= 100
l1 和 l2 均按 非递减顺序 排列
对于python3的链表,题目中有一段提示
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
我们知道了链表包含了两个元素,一个是int型的 val(默认值是0),另一个是next,用来指向下一个结点,所以应该是ListNode型
思路的话可以采用双指针加循环的方式
l1 和 l2 是两个链表,但本质是链表的头指针,所以直接当作指针就行,不需要定义新的指针
比较l1和l2的值,如果l1比l2小,那么将p.next指向l1,然后l1后移(l1 = l1.next);否则p.next指向l2,l2后移
只要有一个表先被遍历完,即 l = None,则退出循环,将没被遍历完的表直接在p的尾巴后面接上
考虑特殊情况,如果l1(l2)一开始就是空表,那么直接返回l2(l1)。
循环中只把两个表遍历了一遍,时间复杂度为O(m+n)
## python3
class Solution:
def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
if l1 == None: # l1为空直接返回l2
return l2
if l2 == None: # l2为空直接返回l1
return l1
result = ListNode() # 存放结果
p = result # 指针
while l1 != None and l2 != None:
if l1.val < l2.val: # 比较两个结点的值,l1的值小
p.next = l1 # 将p的下一个结点指向当前的l1
l1 = l1.next # l1后移
else: # 否则将p的下一个指向当前的l2,l2则后移
p.next = l2
l2 = l2.next
p = p.next
if l1 != None: # 如果l1没有遍历完,则直接在p的后面接上剩余部分
p.next = l1
if l2 != None:
p.next = l2
return result.next
接上继续分析,可以利用递归进行操作
当p指向的是l1的头结点时,问题就变成了l1.next和l2两个有序链表进行合并,问题规模变小了,但问题还是链表的合并,这就意味着可以用递归,代码如下。
## python3
class Solution:
def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
if l1 == None: # l1为空直接返回l2
return l2
if l2 == None: # l2为空直接返回l1
return l1
if l1.val < l2.val:
l1.next = self.mergeTwoLists(l1.next,l2)
return l1
else:
l2.next = self.mergeTwoLists(l1,l2.next)
return l2