题目
给你两个链表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(n+m) O(n+m)。
- 空间复杂度。不需要额外的存储空间因此空间复杂度为 O ( 1 ) O(1) O(1)。