【Leetcode】21. 合并两个有序链表(Merge Two Sorted Lists)

No21. 合并两个有序链表

题目

将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。

示例 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 均按 非递减顺序 排列

解题代码(Python3)

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:
            p = head = l1
            l1 = l1.next
        else:
            p = head = l2
            l2 = l2.next
        while l1 and l2:
            if l1.val <= l2.val:
                head.next = l1
                l1 = l1.next
            else:
                head.next = l2
                l2 = l2.next                
            head = head.next
        else:
            #不可能同时为空
            if l1:
                head.next = l1
            else:
                head.next = l2
        return p

思路:

第一步,如果l1或l2为空,那么返回l2或l1;
第二步,新建p和head为头指针,然后根据大小分别对l1和l2进行连接;
第三步,将l1和l2剩余的部分接入p中,注意此时由于跳出循环,l1和l2只有一个非空。

复杂度分析:

  • 时间复杂度O(n)
  • 空间复杂度O(1)

运行结果:

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/Xiao_Spring/article/details/113731120