题目:
将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
我的方法:
- 第一眼看见就觉得和Day2的题目很像啊,都是链表,不过上一个是将两个链表加起来,这个是比较大小。
dummy=curr=ListNode()
,中途只更改curr(curr.next = ListNode(x)),最后返回dummy.next。- 因为是比较大小,所以用到了前几天学到的
float('inf')
正无穷,当走到节点末尾时,就赋值正无穷。
看了答案之后,原来还可以用递归的思路解。
class Solution:
def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
curr = dummy = ListNode()
# 还没走到末尾
while l1 or l2:
x = l1.val if l1 else float("inf")
y = l2.val if l2 else float("inf")
if (x<=y):
curr.next = ListNode(x)
if l1: l1 = l1.next
else:
curr.next = ListNode(y)
if l2: l2 = l2.next
curr = curr.next
return dummy.next
参考答案 1:
思路跟我的是一样的~
不过while条件用的是and而不是or,因为只要有一个列表到终点了,剩下的直接填到末尾就好了,不用再循环了。
class Solution:
def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
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
递归思路:
官方还给出了一个递归的思路。
- 终止条件:如果 l1 或者 l2 一开始就是空链表 ,那么没有任何操作需要合并,所以我们只需要返回非空链表。
- 否则,我们要判断 l1 和 l2 哪一个链表的头节点的值更小,然后递归地决定下一个添加到结果里的节点。
- 如果两个链表有一个为空,递归结束。
只看代码还是很懵,这个流程画的图很形象~
class Solution:
def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
if l1 is None:
return l2
elif l2 is None:
return l1
# l1更小,就继续合并l1后面的和l2
elif l1.val < l2.val:
l1.next = self.mergeTwoLists(l1.next, l2)
return l1
else:
l2.next = self.mergeTwoLists(l1, l2.next)
return l2