题意:
给你两个链表 list1
和 list2
,它们包含的元素分别为 n
个和 m
个。
请你将 list1
中第 a
个节点到第 b
个节点删除,并将list2
接在被删除节点的位置。
解题思路:
代码详解
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* mergeInBetween(ListNode* list1, int a, int b, ListNode* list2) {
if (list1 == NULL) {
return NULL;
}
ListNode *head = list1, *node; // 定义头节点和中间节点
while (list1->next->val != a) { // 当list1的下一个节点的值等于a时,退出
list1 = list1->next;
}
list1->next = list2; // list1->next 等于list2
node = list1;
while (node->val != b) { // 找到 b
node = node->next;
}
while (list2->next != nullptr) {
list2 = list2->next;
}
list2->next = node->next; // list2的最后一个指针的下一个等于node->下一个
return head;
}
};