题目
给你两个单链表的头节点 headA 和 headB ,请你找出并返回两个单链表相交的起始节点。如果两个链表没有交点,返回 null 。
思路
如果说两个链表相交的话,那么他们以后的部分一定是等长的。
那么我们可以把两个链表从尾部对齐,然后从短的开始看是否相交
代码
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
int GetLen(ListNode *head);
class Solution
{
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB)
{
int Len1 = GetLen(headA);
int Len2 = GetLen(headB);
int Len3;
if(Len1 > Len2)
{
Len3 = (Len1 - Len2);
//尾部对齐
while(Len3--)
{
headA = headA->next;
}
}
else
{
Len3 = (Len2 - Len1);
//尾部对齐
while(Len3--)
{
headB = headB->next;
}
}
while(NULL != headA)
{
//看是否相交
if(headA == headB)
{
return headA;
}
headA = headA->next;
headB = headB->next;
}
return NULL;
}
};
int GetLen(ListNode *head)
{
int Len = 0;
while(head)
{
Len ++;
head = head->next;
}
return Len;
}