LeetCode刷题笔记第21题:合并两个有序链
将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例 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 均按 非递减顺序 排列
想法:
创建一个新的链表的头结点,从两个有序链表的第一个元素开始对比,将值小的赋值给新的链表,循环遍历对比两个有序链表,当有一个有序链表为空时,将另一个链表的剩余部分赋给新的链表。
# 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:
# 设置链表的头结点
head = ListNode(0)
# 将头结点赋值给变量
first = head
# 当链表l1和l2同时不为空时,循环
while l1 and l2:
# 当l1的值大于l2的值时将l2赋给头结点的下一个位置,并将l2的下一个位置赋给当前位置
if l1.val > l2.val:
head.next = l2
l2 = l2.next
# 否则,将l1赋给头结点的下一个位置,并将l1的下一个位置赋给当前位置
else :
head.next = l1
l1 = l1.next
# 头结点得到赋值,将下一个节点赋给当前节点
head = head.next
# 当l1为空时,将l2剩余的部分赋给头结点
if l1 == None:
head.next = l2
# 当l2为空时,将l2剩余的部分赋给头结点
elif l2 == None:
head.next = l1
# 返回链表
return first.next