合并两个链表
难度:中等
给你两个链表 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 接在该位置。上图中蓝色的边和节点为答案链表。
示例 2:
输入:list1 = [0,1,2,3,4,5,6], a = 2, b = 5, list2 = [1000000,1000001,1000002,1000003,1000004]
输出:[0,1,1000000,1000001,1000002,1000003,1000004,6]
解释:上图中蓝色的边和节点为答案链表。
模拟
思路:
题目要求将
list
1
\textit{list}_1
list1 的第
a
a
a 到
b
b
b 个节点都删除,将其替换为
list
2
\textit{list}_2
list2。因此,我们首先找到
list
1
\textit{list}_1
list1 中第
a
−
1
a - 1
a−1 个节点,以及第
b
+
1
b + 1
b+1 个节点。由于
1
≤
a
≤
b
<
n
−
1
1 \le a \le b \lt n - 1
1≤a≤b<n−1(其中
n
n
n 是
list
1
\textit{list}_1
list1 的长度),所以
a
−
1
a-1
a−1 个节点和
b
+
1
b+1
b+1 个节点是一点存在的。
然后我们让第 a − 1 a-1 a−1 个节点的 next \textit{next} next 指向 list 2 \textit{list}_2 list2 的头节点,再让 l i s t 2 list2 list2 的尾节点的 next \textit{next} next 指向第 b + 1 b+1 b+1 个节点即可。
复杂度分析:
- 时间复杂度: O ( n + m ) O(n + m) O(n+m),其中 n n n 是 list 1 \textit{list}_1 list1 的长度, m m m 是 list 2 \textit{list}_2 list2 的长度。
- 空间复杂度: O ( 1 ) O(1) O(1)。
class Solution:
def mergeInBetween(self, list1: ListNode, a: int, b: int, list2: ListNode) -> ListNode:
list3 = list1
for i in range(a - 1):
list1 = list1.next
list4 = list1
for i in range(b - a + 2):
list4 = list4.next
list1.next = list2
while list2.next:
list2 = list2.next
list2.next = list4
return list3
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/merge-in-between-linked-lists