leetcode之链表十三: 1669-合并两个链表

原题: https://leetcode-cn.com/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 中第三和第四个节点，并将 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 <= 104
• 1 <= a <= b < list1.length - 1
• 1 <= list2.length <= 104

二、解题

package com.leetcode.test.linklist;

/**
 * 给你两个链表 list1 和 list2 ，它们包含的元素分别为 n 个和 m 个。

 请你将 list1 中第 a 个节点到第 b 个节点删除，并将list2 接在被删除节点的位置。
 */
public class Solution15 {
    public static void main(String[] args) {
        ListNode a1 = new ListNode(0);
        ListNode a2 = new ListNode(1);
        ListNode a3 = new ListNode(2);
        ListNode a4 = new ListNode(3);
        ListNode a5 = new ListNode(4);
        ListNode a6 = new ListNode(5);
        a1.next = a2;
        a2.next = a3;
        a3.next = a4;
        a4.next = a5;
        a5.next = a6;

        ListNode b1 = new ListNode(1000000);
        ListNode b2 = new ListNode(1000001);
        ListNode b3 = new ListNode(1000002);
        b1.next = b2;
        b2.next = b3;

        ListNode l = mergeInBetween(a1,3,4,b1);

        while (l != null) {
            System.out.print(l.val + " ");
            l = l.next;
        }
    }

    /**
     * 先遍历第一个链表到a的位置，将他的next指向第二个链表
     * 然后同时遍历第一个和第二个链表
     * 第一个链表遍历到b的位置停止
     * 第二个链表遍历到尾结点停止
     * 再将第二个链表的尾结点的next指向第一个链表b位置节点的next，然后将b节点的next设置为null
     *
     */
    public static ListNode mergeInBetween(ListNode list1, int a, int b, ListNode list2) {
        ListNode list1Cur = list1;          //第一个链表当前节点
        ListNode list2Cur = list2;          //第二个链表当前节点
        int pos = 1;                //第一个链表的当前位置

        //先遍历第一个链表到a的位置，将他的next指向第二个链表
        while (pos < a){
            list1Cur = list1Cur.next;
            pos ++;
        }
        ListNode list1Curs = list1Cur.next;          //新建一个指针，用于继续遍历原来的第一个链表
        list1Cur.next = list2Cur;

        //继续遍历第一个链表，直到b的位置停止
        while (pos < b){
            list1Curs = list1Curs.next;
            pos ++;
        }

        //遍历第二个链表直到尾结点
        while (list2Cur.next != null){
            list2Cur = list2Cur.next;
        }

        //再将第二个链表的尾结点的next指向第一个链表b位置节点的next，然后将b节点的next设置为null
        list2Cur.next = list1Curs.next;
        list1Curs.next = null;

        return list1;
    }
}

三、运行结果

四、提交结果

这次结果不错，时间和空间都很好