LeetCode-1669 合并两个链表

题目描述

给你两个链表 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;
    }
}