[剑指offer-52]两个链表的第一个公共节点

[52]两个链表的第一个公共节点

题目

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

当不存在公共节点时，返回空节点。

思路

1. 先计算出两个链表的长度，可以让比较长的先走两个链表长度之差的步数，两个再一起走。
2. 不同部分为a， 和b，公共部分为c；a + c + b = b + c + a；让两个一起走，a走到头就转向b， b走到头转向a，则在公共部分相遇。
3. 遍历链表 A， 将 A 中节点对应的指针（地址），插入 set。再遍历链表 B，将 B 中节点对应的指针（地址），在 set 中查找，发现在 set中的第一个节点地址，即是两个链表的交点。

代码

class Solution {
public:
    ListNode *findFirstCommonNode(ListNode *headA, ListNode *headB) 
    { // 方法2
        
        ListNode* pA = headA;
        ListNode* qB = headB;
        
        while(pA != qB)
        {
            if(pA != NULL) pA = pA->next;
            else pA = headB;
            
            if(qB != NULL) qB = qB->next;
            else qB = headA;
        }
        
        return pA;
    }
};