找到两个单链表相交的起始节点。
注意:
- 如果两个链表没有交点,返回
null
. - 在返回结果后,两个链表仍须保持原有的结构。
- 可假定整个链表结构中没有循环。
- 程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。
联系之前做过有环链表,把链表A的尾段指向链表B的头端,构成一个有环链表。利用之前推导的距离公式,找到相交的节点。
ListNode* Solution::getIntersectionNode(ListNode *headA, ListNode *headB)
{
/*类比之前做过的环形链表*/
if((headA == NULL) || (headB == NULL) )
{
return NULL;
}
/*1.找到headA,headB的尾部节点*/
ListNode *ptail = headB;
while(ptail->next != NULL)
{
ptail = ptail->next;
}
/*2.尾部指向headB的头结点,构成回环*/
ptail->next = headB;
/*3.快慢指针走到相遇处*/
ListNode *pslow = headA;
ListNode *pfast = headA;
while((pfast != NULL) && (pfast->next != NULL))
{
pslow = pslow->next;
pfast = pfast->next->next;
if(pslow == pfast)
{
/*4.新建一个指向headA的头结点,同时移动该节点和pslow,根据距离差应该在交点相会*/
pslow = headA;
while(pslow != pfast)
{
pslow = pslow->next;
pfast = pfast->next;
}
ptail->next = NULL;
return pfast;
}
}
ptail->next = NULL;
return NULL;
}