剑指 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;
}
};