题目链接:https://leetcode.com/problems/intersection-of-two-linked-lists/
Write a program to find the node at which the intersection of two singly linked lists begins.
For example, the following two linked lists:
A: a1 → a2 ↘ c1 → c2 → c3 ↗ B: b1 → b2 → b3
begin to intersect at node c1.
Notes:
- If the two linked lists have no intersection at all, return
null
. - The linked lists must retain their original structure after the function returns.
- You may assume there are no cycles anywhere in the entire linked structure.
- Your code should preferably run in O(n) time and use only O(1) memory.
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
if(headA == NULL || headB == NULL)
return NULL;
ListNode* iter1 = headA;
ListNode* iter2 = headB;
size_t len1 = 1;
while(iter1->next != NULL)
{
iter1 = iter1->next;
len1 ++;
}
size_t len2 = 1;
while(iter2->next != NULL)
{
iter2 = iter2->next;
len2 ++;
}
if(iter1 != iter2)
return NULL;
if(len1 > len2)
{
size_t dif=len1-len2;
while(dif>0){
headA = headA->next;
dif--;
}
}
else if(len2 > len1)
{
size_t dif=len2-len1;
while(dif>0){
headB = headB->next;
dif--;
}
}
while(headA != headB)
{
headA = headA->next;
headB = headB->next;
}
return headA;
}
};