剑指 Offer 52. 两个链表的第一个公共节点

剑指 Offer 52. 两个链表的第一个公共节点

输入两个链表，找出它们的第一个公共节点。

示例 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 个节点。
==============================================================

class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        if (headA == NULL || headB == NULL)
            return NULL;
        int n = 0;
        int m = 0; 
        ListNode *A = headA;
        ListNode *B = headB;
        while (A) {
            n++;
            A = A->next;
        }
        while (B) {
            m++;
            B = B->next;
        }
        while(n>m&&headA) {
            headA = headA->next;
            n--;
        }
        while (n < m&&headB) {
            headB = headB->next;
            m--;
        }
        while (headA != headB&&headA&&headB) {
            headA = headA->next;
            headB = headB->next;
        }
        if (headA == headB)
            return headA;
        else
            return NULL;
    }
};

很罗嗦，算法的执行效率不高，去看了题解，跪着看的

class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        if (headA == NULL || headB == NULL)
            return NULL;
    
        ListNode *A = headA;
        ListNode *B = headB;
        while (A != B) {
            A = A == NULL ? headB : A->next;
            B = B == NULL ? headA : B->next;
        }
        return A;
    }
};