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

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

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

思路

思路一：使用 map 记录已经访问的节点

思路二：使用双指针，可以将空间复杂度将至 O(1)

设 A, B 两个链表，公共节点前的长度分别为 a, b, 公共节点的长度为 c，则 A 的长度为a+c， B 的长度为 b+c

如果两链表相交，一定会返回相交节点 a + c + b = b + c + a

如果两链表不相交，则 c 为0

a + b = b + a，最后会返回 nullptr

代码

class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        if (headA == nullptr || headB == nullptr) return nullptr;

        ListNode *pA = headA, *pB = headB;
        while (pA != pB) {
            pA = pA == nullptr ? headB : pA->next;
            pB = pB == nullptr ? headA : pB->next;
        }
        return pA;
        
    }
};