题目
来源:JZ52 两个链表的第一个公共结点
描述
输入两个无环的单向链表,找出它们的第一个公共结点,如果没有公共节点则返回空。(注意因为传入数据是链表,所以错误测试数据的提示是用其他方式显示的,保证传入数据是正确的)
数据范围: n≤1000
要求:空间复杂度 O(1),时间复杂度 O(n)
例如,输入{1,2,3},{4,5},{6,7}时,两个无环的单向链表的结构如下图所示:
可以看到它们的第一个公共结点的结点值为6,所以返回结点值为6的结点。
输入描述:
输入分为是3段,第一段是第一个链表的非公共部分,第二段是第二个链表的非公共部分,第三段是第一个链表和第二个链表的公共部分。 后台会将这3个参数组装为两个链表,并将这两个链表对应的头节点传入到函数FindFirstCommonNode里面,用户得到的输入只有pHead1和pHead2。
返回值描述:
返回传入的pHead1和pHead2的第一个公共结点,后台会打印以该节点为头节点的链表。
示例1
输入:
{1,2,3},{4,5},{6,7}
返回值:
{6,7}
说明:
第一个参数{1,2,3}代表是第一个链表非公共部分,第二个参数{4,5}代表是第二个链表非公共部分,最后的{6,7}表示的是2个链表的公共部分
这3个参数最后在后台会组装成为2个两个无环的单链表,且是有公共节点的
示例2
输入:
{1},{2,3},{}
返回值:
{}
说明:
2个链表没有公共节点 ,返回null,后台打印{}
解析
图已经很明确了,如果有公共部分,两个链表是交叉的。最直观的想法是用两个指针分别扫描两个链表,当遇到相同节点时,当前节点即公共部分的头节点,但是这里要注意几点:
1.公共节点指节点地址相同,而不是节点的值相同
2.公共部分可能是没有的
3.非公共部分长度未必一致
第一点不用说,用指针比较即可,第二点不用处理,当前指针都是空指针时就代表无公共部分。最后一点也不难处理,先扫描两个链表统计长度,根据长度差,较长的链表先走一段长度差,这样两个指针位置就对齐了。
综上,算法是:设两个指针,分别扫描两个链表,统计各自长度。然后根据长度差,较长链表的指针先走一段长度差,最后两个指针平行移动,当两个指针相同时即公共部分的头节点(无论是否为空)。
ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
ListNode*p = pHead1, *q = pHead2;
int len1 = 0, len2 = 0;
while (p) {
len1++;
p = p->next;
}
while (q) {
len2++;
q = q->next;
}
p = pHead1, q = pHead2;
if (len1 < len2) {
while (len1 != len2) {
len2--;
q = q->next;
}
} else if (len2 < len1) {
while (len1 != len2) {
len1--;
p = p->next;
}
}
while (p!=q) {
p = p->next;
q = q->next;
}
return p;
}