No21. 合并两个有序链表
题目
将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例 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)
class Solution:
def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
if not l1:
return l2
if not l2:
return l1
if l1.val <= l2.val:
p = head = l1
l1 = l1.next
else:
p = head = l2
l2 = l2.next
while l1 and l2:
if l1.val <= l2.val:
head.next = l1
l1 = l1.next
else:
head.next = l2
l2 = l2.next
head = head.next
else:
#不可能同时为空
if l1:
head.next = l1
else:
head.next = l2
return p
思路:
第一步,如果l1或l2为空,那么返回l2或l1;
第二步,新建p和head为头指针,然后根据大小分别对l1和l2进行连接;
第三步,将l1和l2剩余的部分接入p中,注意此时由于跳出循环,l1和l2只有一个非空。
复杂度分析:
- 时间复杂度O(n)
- 空间复杂度O(1)