LeetCode 160. Intersection of Two Linked Lists

Tag：List

Difficulty：Easy

Problem

 相交链表
 给你两个单链表的头节点 headA 和 headB ，请你找出并返回两个单链表相交的起始节点。如果两个链表不存在相交节点，返回 null 。
 图示两个链表在节点 c1 开始相交：

 题目数据 保证 整个链式结构中不存在环。
 注意，函数返回结果后，链表必须 保持其原始结构 。

 自定义评测：
 评测系统 的输入如下（你设计的程序 不适用 此输入）：
 intersectVal - 相交的起始节点的值。如果不存在相交节点，这一值为 0
 listA - 第一个链表
 listB - 第二个链表
 skipA - 在 listA 中（从头节点开始）跳到交叉节点的节点数
 skipB - 在 listB 中（从头节点开始）跳到交叉节点的节点数
 评测系统将根据这些输入创建链式数据结构，并将两个头节点 headA 和 headB 传递给你的程序。如果程序能够正确返回相交节点，那么你的解决方案将被 视作正确答案 。

Example

输入：intersectVal = 8, listA = [4,1,8,4,5], listB = [5,6,1,8,4,5], skipA = 2, skipB = 3
输出：Intersected at '8'

Solution

解法一：

 1. 遍历2个链表，记录链表长度。
 2. 如果两个链表尾节点不同，则一定不相交。
 3. 否则，计算两链表长度差值。较长的链表先走差值步，然后两链表一起走，相遇即为交叉点

public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
    if (headA == null || headB == null ){
        return null;
    }
    ListNode a = headA, b = headB;      // 找结束节点
    int lengthA = 0, lengthB = 0;       // 计算总数
    while (a != null) {
        lengthA++;
        a = a.next;
    }
    while (b != null) {
        lengthB++;
        b = b.next;
    }
    // a != b 说明尾节点不同 不相交
    if (a != b) {
        return null;
    }

    // 相交 长链表先走差值步，然后一起走, 相遇
    a = lengthA - lengthB > 0 ? headA : headB;   // 指向长链表
    b = a == headA ? headB : headA;              // 指向短链表
    lengthA = Math.abs(lengthA - lengthB);       // 差值，其实只用一个变量就好，因为只要差值

    // 长链表先走差值步
    while (lengthA != 0) {
        a = a.next;
        lengthA--;
    }

    // 一起走
    while (a != b) {
        a = a.next;
        b = b.next;
    }
    return a;

}

解法二：

讨论区学来的，非常巧妙的方法。

链表中不存在环，但是可以用环的思想来做。两条链表分别从各自的开头开始往后遍历，当其中一条遍历到末尾时，跳到另一个条链表的开头继续遍历。两个指针最终会相等，而且只有两种情况，一种情况是在交点处相遇，另一种情况是在各自的末尾。

（想不出来的方法

public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
    if (headA == null || headB == null) {
        return null;
    }
    ListNode a = headA, b = headB;

    //if a & b have different len, then we will stop the loop after second iteration
    while( a != b){
        //for the end of first iteration, we just reset the pointer to the head of another linkedlist
        a = a == null? headB : a.next;
        b = b == null? headA : b.next;
    }

    return a;
}