LeetCode-1669 合并两个链表

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/merge-in-between-linked-lists

题目描述

给你两个链表 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]
解释：上图中蓝色的边和节点为答案链表。

提示：

3 <= list1.length <= 10⁴
1 <= a <= b < list1.length - 1
1 <= list2.length <= 10⁴

解题思路

这题是一道很简单链表数据结构题，解题的关键是需要记录删除结点的前一个结点和删除结点的后一个结点以便之后的连接。

代码展示

/**
 * 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) {
        ListNode *preA = nullptr, *eveB = nullptr, *p = list1;
        int iCount = 0;
        while(p)
        {
            ListNode *q = nullptr;
            if(iCount == a - 1)
                preA = p;
            if(iCount == b + 1)
                eveB = p;
            if(iCount >= a && iCount <= b)
                q = p;
            p = p->next;
            iCount++;
            if(q)
                delete q;
        }
        preA->next = list2;
        ListNode * l = list2;
        while(l->next)
            l = l->next;
        l->next = eveB;
        return list1;
    }
};

运行结果