一、题目描述
将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例:
输入:1->2->4, 1->3->4
输出:1->1->2->3->4->4
二、解题思路 & 代码
2.1 递归
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
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):
l1.next = self.mergeTwoLists(l1.next, l2)
return l1
else:
l2.next = self.mergeTwoLists(l1, l2.next)
return l2
复杂度分析
-
时间复杂度:
O(n+m)
,其中n
和m
分别为两个链表的长度。因为每次调用递归都会去掉l1
或者l2
的头节点(直到至少有一个链表为空),函数mergeTwoList
至多只会递归调用每个节点一次。因此,时间复杂度取决于合并后的链表长度,即O(n+m)
。 -
空间复杂度:
O(n+m)
,其中n
和m
分别为两个链表的长度。递归调用mergeTwoLists
函数时需要消耗栈空间,栈空间的大小取决于递归调用的深度。结束递归调用时mergeTwoLists
函数最多调用n+m
次,因此空间复杂度为O(n+m)
2.2 非递归
首先,我们设定一个哨兵节点 prehead
,这可以在最后让我们比较容易地返回合并后的链表。我们维护一个 prev
指针,我们需要做的是调整它的 next
指针。然后,我们重复以下过程,直到 l1
或者 l2
指向了 null
:如果 l1
当前节点的值小于等于 l2
,我们就把 l1
当前的节点接在 prev
节点的后面同时将 l1
指针往后移一位。否则,我们对 l2
做同样的操作。不管我们将哪一个元素接在了后面,我们都需要把 prev
向后移一位。
在循环终止的时候, l1
和 l2
至多有一个是非空的。由于输入的两个链表都是有序的,所以不管哪个链表是非空的,它包含的所有元素都比前面已经合并链表中的所有元素都要大。这意味着我们只需要简单地将非空链表接在合并链表的后面,并返回合并链表即可。
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def mergeTwoLists(self, l1, l2):
prehead = ListNode(-1)
prev = prehead
while l1 and l2:
if l1.val <= l2.val:
prev.next = l1
l1 = l1.next
else:
prev.next = l2
l2 = l2.next
prev = prev.next
# 合并后 l1 和 l2 最多只有一个还未被合并完,我们直接将链表末尾指向未合并完的链表即可
prev.next = l1 if l1 is not None else l2
return prehead.next
复杂度分析
-
时间复杂度:O(n+m) ,其中 n 和 m 分别为两个链表的长度。因为每次循环迭代中,l1 和 l2 只有一个元素会被放进合并链表中, 因此 while 循环的次数不会超过两个链表的长度之和。所有其他操作的时间复杂度都是常数级别的,因此总的时间复杂度为 O(n+m)。
-
空间复杂度:O(1) 。我们只需要常数的空间存放若干变量。