题目描述
输入两个无环的单链表,找出它们的第一个公共结点。(注意因为传入数据是链表,所以错误测试数据的提示是用其他方式显示的,保证传入数据是正确的)
解法
首先考虑特殊情况,两链表有一个为空,返回空指针
分别求两个链表的长度,那么有可能的第一个公共结点是:长链表走到从长链表结点数-短链表结点数的那个结点,然后短链表从头开始,两链表依次走一个结点,比较是否相同
class Solution {
public:
ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
if(pHead1 == pHead2) return pHead1;
if(pHead1 == NULL || pHead2 == NULL) return nullptr;
ListNode* p1, *p2;
p1 = pHead1;
p2 = pHead2;
int i = 0, j = 0;
//获取两个链表的长度
while(p1 != NULL)
{
i++;
p1 = p1 ->next;
}
while(p2 != NULL)
{
j++;
p2 = p2 ->next;
}
int k;
if(i >= j) k = i -j;
else k = j - i;
p1 = pHead1;
p2 = pHead2;
//让更长的链表走到第一个可能公共结点处
if(i >= j)
{
i = 0;
while(i < k)
{
i++;
p1 = p1->next;
}
while(p1 != p2 && p1 != NULL && p2 != NULL)
{
p1 = p1->next;
p2 = p2->next;
}
return p1;
}
else
{
j = 0;
while(j < k)
{
j++;
p2 = p2->next;
}
while(p1 != p2 && p1 != NULL && p2 != NULL)
{
p1 = p1->next;
p2 = p2->next;
}
return p1;
}
}
};
时间复杂度O(n)
空间复杂度O(1)