一、问题:
链接:https://leetcode-cn.com/problems/merge-two-sorted-lists/
将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例 1:
输入:l1 = [1,2,4], l2 = [1,3,4],输出:[1,1,2,3,4,4]
示例 2:
输入:l1 = [], l2 = [],输出:[]
示例 3:
输入:l1 = [], l2 = [0],输出:[0]
提示:
1. 两个链表的节点数目范围是 [0, 50]
2. -100 <= Node.val <= 100
3. l1 和 l2 均按 非递减顺序 排列
二、代码:
# 方法一:迭代法
# 思路:
# 1. 创建一个新链表,依次比较两个链表的指针指向元素的大小,将新链表的指针指向较小元素,并移动相关指针;
# 2. 若其中一个链表遍历完成,则将另一个链表直接接到新链表最后。
# 时间复杂度:O(n+m)
# 空间复杂度:O(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:
prehead = ListNode(0) # 定义链表
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
if l1:
prev.next = l1
else:
prev.next = l2
return prehead.next
# 方法二:递归法。
# 思路:每个子步骤都有相同的任务,因此可以用递归。
# 1. 如果 l1 的元素更小,那么就用 l1.next 和 l2 作比较;反之同理。
# 2. 递归结束条件:
# 1)l1 为空,那么返回 l2;
# 2)l2 为空,那么返回 l1;
# 3)若 l1 的元素较小,那么保留 l1 已有的,并在 l1 后面接上 l1.next 和 l2 合并后的有序链表,且返回 l1。
# 4)若 l2 的元素较小,那么保留 l2 已有的,并在 l2 后面接上 l2.next 和 l1 合并后的有序链表, 且返回 l2。
# 时间复杂度:O(n+m)
# 空间复杂度:O(n+m)
class Solution:
def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
if not l1:
return l2
elif 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
三、知识点:
1. ListNode 新建链表,元素就是括号里面的元素。而 p/prev 只是链表的头指针,指针仅指向一个元素。如果要返回链表结果,要用 prehead.next~