leetcode 1669. 合并两个链表【python3一次遍历方法详细过程描述】

题目

给你两个链表list1和list2，它们包含的元素分别为n个和m个。请你将list1中下标从a到b的全部节点都删除，并将list2接在被删除节点的位置。
下图中蓝色边和节点展示了操作后的结果：

请你返回结果链表的头指针。

示例1：

• 输入：list1 = [0,1,2,3,4,5], a = 3, b = 4, list2 = [1000000,1000001,1000002]
• 输出：[0,1,2,1000000,1000001,1000002,5]
• 解释：我们删除 list1 中下标为 3 和 4 的两个节点，并将 list2 接在该位置。上图中蓝色的边和节点为答案链表。

题解

题目很简单，即把list1中的a到b这一部分用list2替换。思路也很简单先找到list1上的a-1节点，把共next指针指向list2，然后遍历到list2的末尾，再找到b+1节点，把list2的末尾指针的next指针指向list1的b+1。下面以图例来说明整个过程。
第一步，遍历list1到a-1节点。这一步需要切断a-1到a的链接，把a-1的next指针指向list2的首个节点，因此这里需要一个变量来记录a节点。

第二步pre指针遍历list2真到末尾，curr的指针遍历到b+1位置，然后让pre的next指针指向curr。这样就完成了两个链表的合并。

基于以上的过程自写代码：

class Solution:
    def mergeInBetween(self, list1: ListNode, a: int, b: int, list2: ListNode) -> ListNode:
        result=ListNode(0,None)
        result.next=list1
        preA=result
        curr=list1
        for _ in range(a):
            preA=preA.next
            curr=curr.next
        #list2
        preA.next=list2
        while preA.next:
            preA=preA.next
        #curr走到b
        for _ in range(a,b+1):
            curr=curr.next
        preA.next=curr
        return result.next

发现官网代码写的还是比较精炼，思路一样。

class Solution:
    def mergeInBetween(self, list1: ListNode, a: int, b: int, list2: ListNode) -> ListNode:
        preA = list1
        for _ in range(a - 1):
            preA = preA.next
        preB = preA
        for _ in range(b - a + 2):
            preB = preB.next
        preA.next = list2
        while list2.next:
            list2 = list2.next
        list2.next = preB
        return list1

计算复杂性

• 时间复杂度。两个链表都需要遍历一次，因此时间复杂度为$O(n+m)$。
• 空间复杂度。不需要额外的存储空间因此空间复杂度为$O(1)$。