题目
将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例 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 均按 非递减顺序 排列
题目链接
我的思路
- 处理空链表
- 设置一个头节点cur和最后用来返回的res,cur = res = ListNode(0)
- 用一个循环,当两个链表都不为空时,每次比较两个链表,将小的那个作为cur.next,同时它自己往后移,取自己的next
- 当有一个链表为空时退出循环,用两个判断来判断两个链表哪个不为空,将不为空的那个(如果有)接到cur后面
- 返回res.next,因为res是自己定义的头节点,它的next才是结果
我的代码
# 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, list1: Optional[ListNode], list2: Optional[ListNode]) -> Optional[ListNode]:
if not list1: return list2
if not list2: return list1
cur = ListNode(0)
res = cur
while list1 and list2:
if list1.val < list2.val:
cur.next = list1
list1 = list1.next
else:
cur.next = list2
list2 = list2.next
cur=cur.next
if list1:
cur.next = list1
else:
cur.next = list2
return res.next
题解思路
题解还有另一种思路:递归
原问题可以转化成每次只比较两个节点的大小:
- 如果list1的当前节点值小于list2的当前节点值,说明list1的当前节点应该在合并后的链表中排在前面。那么将list1的当前节点作为合并后链表的当前节点,然后递归地合并list1的下一个节点和整个list2,最后返回list1
- 如果list2的当前节点值小于或等于list1的,那么将list2的当前节点作为合并后链表的当前节点,然后递归地合并list1和list2的下一个节点。最后返回list2
边界处理:
- 当list1或list2为空时,返回不为空的那个作为链表的尾巴
参考代码
# 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, list1: Optional[ListNode], list2: Optional[ListNode]
) -> Optional[ListNode]:
if not list1:
return list2
if not list2:
return list1
if list1.val < list2.val:
list1.next = self.mergeTwoLists(list1.next, list2)
return list1
list2.next = self.mergeTwoLists(list1, list2.next)
return list2
Q&A
- 递归解法中为什么写
list1 = self.mergeTwoLists(list1.next, list2)
会出错?list1.next = self.mergeTwoLists(list1.next, list2)
:将list1的下一个节点(list1.next)替换为self.mergeTwoLists(list1.next, list2)的返回值,即合并list1.next和list2后的新链表的头节点。这样做保留了list1节点在原始链表中的位置,只是更新了它的next指针,指向了合并后的链表。list1 = self.mergeTwoLists(list1.next, list2)
:将list1变量本身替换为self.mergeTwoLists(list1.next, list2)的返回值,即合并list1.next和list2后的新链表的头节点。这样做会导致list1变量失去了对原始链表中第一个节点的引用,因为变量被重新赋值了。这将导致原始的list1节点(如果它的值小于list2的头节点值)无法被正确地包含在合并后的链表中,因为递归函数返回时,无法追溯到这个节点。