题目描述:
给你两个链表 list1 和 list2 ,它们包含的元素分别为 n 个和 m 个。
请你将 list1 中第 a 个节点到第 b 个节点删除,并将list2 接在被删除节点的位置。
下图中蓝色边和节点展示了操作后的结果:
链接:https://leetcode-cn.com/problems/merge-in-between-linked-lists
思路分析:删除list链表的中间节点,再进行链接。
- 找到
list1
删除位置的前一个节点 - 找到
list1
删除位置的最后一个节点 - 释放删除节点(避免内存泄漏)
- 找到
list2
的尾节点,用于链接
代码:
class Solution {
public:
ListNode* mergeInBetween(ListNode* list1, int a, int b, ListNode* list2) {
if(list1 == nullptr)
return nullptr;
ListNode* list1FreeHead = list1;
ListNode* list1FreeTail = list1;
ListNode* list1Link = nullptr;
while(a--)//寻找断开的首节点
{
list1Link = list1FreeHead;
list1FreeHead = list1FreeHead->next;
}
ListNode* list1Break = nullptr;
for(int i = 0; i <= b; ++i)//寻找断开的尾节点
{
list1Break = list1FreeTail;
list1FreeTail = list1FreeTail->next;
}
list1Break->next = nullptr;
while(list1FreeHead != nullptr)//释放节点,避免内存泄漏
{
ListNode* tmp = list1FreeHead;
list1FreeHead = list1FreeHead->next;
delete tmp;
}
list1Link->next = list2;//list1链接上list2的头节点
while(list2 != nullptr)//寻找list尾节点
{
if(list2->next == nullptr)
break;
list2 = list2->next;
}
list2->next = list1FreeTail;//list2的尾链接上释放节点的尾
return list1;//返回
}
};