剑指offer25 合并两个排序的链表 Kotlin

题目描述

输入两个递增的链表，单个链表的长度为n，合并这两个链表并使新链表中的节点仍然是递增排序的。数据范围： 0≤n≤10000≤n≤1000，−1000≤节点值≤1000−1000≤节点值≤1000
要求：空间复杂度 O(1)O(1)，时间复杂度 O(n)O(n)

如输入{1,3,5},{2,4,6}时，合并后的链表为{1,2,3,4,5,6}，所以对应的输出为{1,2,3,4,5,6}，转换过程如下图所示：

或输入{-1,2,4},{1,3,4}时，合并后的链表为{-1,1,2,3,4,4}，所以对应的输出为{-1,1,2,3,4,4}，转换过程如下图所示：

 解题思路

阅读题目后，我们可以得到以下信息和需求

1.两个链表都是递增排序：这意味着我们可以利用它们的有序性，通过比较当前节点的值来决定下一个节点的顺序。

2.合并后的链表也需为递增排序：因此我们需要在合并过程中保持顺序不变。

3.空间复杂度要求 O(1)：我们不能使用大量额外的空间来保存数据，这意味着我们需要在原有链表上进行操作。

简单直观的方法

一种直观的方法是用两个链表的值分别存入数组或列表中，进行排序后再构建新的链表。这种方法需要额外的空间来存储节点值，空间复杂度为 O(n)，不符合题目要求。

递归法

另一种思路是递归合并两个链表：

比较两个链表当前节点的值，选择较小的节点作为合并后的链表的当前节点。对剩余部分递归执行上述操作。但递归方法在处理较长链表时可能会导致栈溢出，并且递归调用本身需要额外的空间，空间复杂度为 O(n)，也无法满足题目要求，因此我们考虑用迭代法解题。

迭代法

题目要求合并两个链表，双指针法一次遍历两个链表，利用双指针分别指向两个链表的当前节点也是非常高效的选择，下面是使用迭代法配合双指针的具体实现

/**
 * class ListNode(var `val`: Int) {
 *     var next: ListNode? = null
 * }
 */
object Solution {
    /**
    * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
    *
    * 
    * @param pHead1 ListNode类 
    * @param pHead2 ListNode类 
    * @return ListNode类
    */
    fun Merge(pHead1: ListNode?, pHead2: ListNode?): ListNode? {
        val dummy = ListNode(0)
        var current = dummy
        var l1 = pHead1
        var l2 = pHead2

        while (l1 != null && l2 != null) {
            if (l1.`val` <= l2.`val`) {
                current.next = l1
                l1 = l1.next
            } else {
                current.next = l2
                l2 = l2.next
            }
            current = current.next!!
        }

        if (l1 != null) {
            current.next = l1
        } else {
            current.next = l2
        }

        return dummy.next
    }
}

合并逻辑

使用一个while循环，当两个链表都非空时继续执行。在每次循环中，比较l1和l2的当前节点的值：如果l1.val小于等于l2.val，则将l1的当前节点连接到current.next，并移动l1到下一个节点。否则，将l2的当前节点连接到current.next，并移动l2到下一个节点。无论哪种情况，current都移动到其新连接的节点。当l1和l2中的一个或两个都为空时，循环结束。如果l1非空，将l2的剩余部分连接到current.next。如果l2非空，将l2的剩余部分连接到current.next。