326 字 1 分钟阅读

给定两个(单向)链表,判定它们是否相交并返回交点。请注意相交的定义基于节点的引用,而不是基于节点的值。换句话说,如果一个链表的第k个节点与另一个链表的第j个节点是同一节点(引用完全相同),则这两个链表相交。

示例 1:

输入:intersectVal = 8, listA = [4,1,8,4,5], listB = [5,0,1,8,4,5], skipA = 2, skipB = 3
输出:Reference of the node with value = 8
输入解释:相交节点的值为 8 (注意,如果两个列表相交则不能为 0)。从各自的表头开始算起,链表 A 为 [4,1,8,4,5],链表 B 为 [5,0,1,8,4,5]。在 A 中,相交节点前有 2 个节点;在 B 中,相交节点前有 3 个节点。

示例 2:

输入:intersectVal = 2, listA = [0,9,1,2,4], listB = [3,2,4], skipA = 3, skipB = 1
输出:Reference of the node with value = 2
输入解释:相交节点的值为 2 (注意,如果两个列表相交则不能为 0)。从各自的表头开始算起,链表 A 为 [0,9,1,2,4],链表 B 为 [3,2,4]。在 A 中,相交节点前有 3 个节点;在 B 中,相交节点前有 1 个节点。

示例 3:

输入:intersectVal = 0, listA = [2,6,4], listB = [1,5], skipA = 3, skipB = 2
输出:null
输入解释:从各自的表头开始算起,链表 A 为 [2,6,4],链表 B 为 [1,5]。由于这两个链表不相交,所以 intersectVal 必须为 0,而 skipA 和 skipB 可以是任意值。
解释:这两个链表不相交,因此返回 null。

注意:

  • 如果两个链表没有交点,返回 null
  • 在返回结果后,两个链表仍须保持原有的结构。
  • 可假定整个链表结构中没有循环。
  • 程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。

思路

分情况讨论:

  • 如果链表中有一个是空的,那么直接返回 null。
  • 如果链表末尾元素不一样,那么说明一定不会相交,所以也返回 null。
  • 除去上面两种情况,就说明链表一定会相交。先计算两个链表的长度,并计算出长度差 n。然后较长的链表先移动 n 位,再两个链表一起移动,如果节点相等则为相交点
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
    if(headA == null || headB == null){
        return null;
    }
    int lengthA = 0;
    int lengthB = 0;
    ListNode nodeA = headA;
    ListNode nodeB = headB;
    //获取ListA长度
    while(nodeA != null){
        nodeA = nodeA.next;
        lengthA++;
    }
    //获取ListB长度
    while(nodeB != null){
        nodeB = nodeB.next;
        lengthB++;
    }
    
    //判断终点是否相等
    if(nodeA != nodeB){
        return null;
    }
    
    //获取两条链表长度差n
    boolean langerA = lengthA >= lengthB;
    int n = langerA ? lengthA - lengthB : lengthB - lengthA;
    
    //较长的链表先移动n位
    while(langerA && n > 0){
        headA = headA.next;
        n--;
    }
    while(!langerA && n > 0){
        headB = headB.next;
        n--;
    }
    
    //两个链表一起移动直到到达终点或者相交
    while(headA != headB){
        headA = headA.next;
        headB = headB.next;
    }
    return headA;
}

另一种思路

假设 A 长度 length(A),B 长度 length(B),A 相交前的长度为 c,B 相交前的长度为 d,那么:length(A) - c = length(B) - d,整理得到:length(A) + d = length(B) + c,这个式子说明我们经过遍历和交换后步伐将会一致,具体可举例模拟,非常巧妙。 当步伐一致后,如果能够遇到相交的第一个节点,则跳出 while,返回 a 指向的相交节点指针;若找不到相交节点,则最终两个指针均走向末尾 null,此时也会跳出 while,并且返回 null。

public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
    ListNode tempA = headA;
    ListNode tempB = headB;
    while (tempA != tempB) {
        tempA = tempA == null ? headB : tempA.next;
        tempB = tempB == null ? headA : tempB.next;
    }
    return tempA;
}