题目描述
给你两个链表 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 接在该位置。上图中蓝色的边和节点为答案链表。
思路
很简单,找到a
之前的那个节点和b
之后的那个节点,将其分别与list2
的首尾相连即可。
代码
class Solution {
public ListNode mergeInBetween(ListNode list1, int a, int b, ListNode list2) {
// 创建虚拟节点作为新链表的起点
ListNode dummy = new ListNode(0);
dummy.next = list1;
ListNode pre = dummy;
int i = 0;
// pre指针向后移动到a的位置
while(i<a){
i++;
pre = pre.next;
}
// tail指向a的下一个位置,即要删除的第一个节点
ListNode tail = pre.next;
// 将pre与list2的首部连接
pre.next = null;
pre.next = list2;
// tail指针向后移动到b的位置,即要删除的最后一个节点
while(i<b){
i++;
tail = tail.next;
}
ListNode cur = list2;
// cur指针找到list2的最后一个节点
while(cur.next!=null){
cur = cur.next;
}
// 将cur与tail后一个节点相连
cur.next = tail.next;
return dummy.next;
}
}