剑指offer52 两个链表的第一个公共结点 Kotlin

描述

输入两个无环的单向链表，找出它们的第一个公共结点，如果没有公共节点则返回空。（注意因为传入数据是链表，所以错误测试数据的提示是用其他方式显示的，保证传入数据是正确的）

数据范围： n≤1000n≤1000  要求：空间复杂度 O(1)O(1)，时间复杂度 O(n)O(n)

例如，输入{1,2,3},{4,5},{6,7}时，两个无环的单向链表的结构如下图所示：

可以看到它们的第一个公共结点的结点值为6，所以返回结点值为6的结点。

输入描述：

输入分为是3段，第一段是第一个链表的非公共部分，第二段是第二个链表的非公共部分，第三段是第一个链表和第二个链表的公共部分。 后台会将这3个参数组装为两个链表，并将这两个链表对应的头节点传入到函数FindFirstCommonNode里面，用户得到的输入只有pHead1和pHead2。

返回值描述：

返回传入的pHead1和pHead2的第一个公共结点，后台会打印以该节点为头节点的链表。

解题思路

题目我们找到两个单向链表的第一个公共结点。题目保证输入的链表是无环的，因此不存在环的干扰。由于链表长度不确定，我们需要一种高效的方式在 O(n) 时间复杂度和 O(1) 空间复杂度内找到这个公共结点。经过剑指offer25 合并两个排序的链表后，我们对于合并两个链表和空间复杂度O(1)两个要求应该立刻想到使用迭代法配合双指针法进行解题。

这种方法的核心思想是：

1.使用两个指针分别遍历两个链表。

2.当一个指针遍历到链表尾部时，将它指向另一个链表的头部。

3.当两个指针相遇时，该节点即为第一个公共节点。

两个指针分别遍历两个链表，并在到达尾部时切换到另一个链表的头部。这样，两指针走过的路径长度相同，即 A链表长度 + B链表长度。如果存在公共结点，它们会在第二次遍历时相遇。

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

        // 当两个指针相等时，即为第一个公共节点
        while (p1 != p2) {
            // 当p1到达链表末尾时，重新定位到pHead2
            p1 = if (p1 == null) pHead2 else p1.next
            // 当p2到达链表末尾时，重新定位到pHead1
            p2 = if (p2 == null) pHead1 else p2.next
        }
        
        return p1 // or return p2, as p1 == p2
    }
}

FindFirstCommonNode方法

首先检查两个链表的头节点是否为null。如果任一链表为空，则直接返回null。使用一个while循环，当两个指针不相等时继续执行：

如果p1到达链表末尾（即p1为null），则将其重新定位到pHead2。

如果p2到达链表末尾（即p2为null），则将其重新定位到pHead1。

p1和p2相等时，循环结束。此时，p1和p2指向的就是两个链表的第一个公共节点。最后，返回p1作为结果。由于p1 == p2，因此返回p2也是正确的。