- 题目:
编写一个程序,找到两个单链表相交的起始节点。
如下面的两个链表:
在节点 c1 开始相交。
示例 1:
输入:intersectVal = 8, listA = [4,1,8,4,5], listB = [5,0,1,8,4,5], skipA = 2, skipB = 3
输出:Reference of the node with value = 8
输入解释:相交节点的值为 8 (注意,如果两个列表相交则不能为 0)。从各自的表头开始算起,链表 A 为 [4,1,8,4,5],链表 B 为 [5,0,1,8,4,5]。在 A 中,相交节点前有 2 个节点;在 B 中,相交节点前有 3 个节点。
- 解题思路一
将链表尾部对齐。然后判断节点是否相等。
代码实现:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
if(headA == nullptr || headB == nullptr)
return nullptr;
int l1 = 0, l2 = 0; //记录两个链表的长度
ListNode *p = headA, *q = headB;
while(p != nullptr) //计算LA的长度
{
l1++;
p = p->next;
}
while(q != nullptr) //计算LB的长度
{
l2++;
q = q->next;
}
p = headA, q =headB;
if(l1 > l2)
{
int k = l1 - l2;
while(k != 0 && p != nullptr)
{
p = p->next;
k--;
}
while(p != nullptr && q != nullptr)
{
if(p == q)
return p;
p = p->next;
q = q->next;
}
}
if(l2 >= l1)
{
int k = l2 - l1;
while(k != 0 && q != nullptr)
{
q = q->next;
k--;
}
while(p != nullptr && q != nullptr)
{
if(p == q)
{
return p;
}
p = p->next;
q = q->next;
}
}
return nullptr;
}
- 解题思路二:(参考其他解答)
双指针法:定义两个指针,由于(a+b).length() = (b+a).length()
且由于后面的节点是公共的节点,所以第二轮遍历第一个相同节点则为公共节点
代码实现:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
/**
定义两个指针, 第一轮让两个到达末尾的节点指向另一个链表的头部, 最后如果相遇则为交点(在第一轮移动中恰好抹除了长度差)
两个指针等于移动了相同的距离, 有交点就返回, 无交点就是各走了两条指针的长度
**/
if(headA == nullptr || headB == nullptr) return nullptr;
ListNode *pA = headA, *pB = headB;
// 在这里第一轮体现在pA和pB第一次到达尾部会移向另一链表的表头, 而第二轮体现在如果pA或pB相交就返回交点, 不相交最后就是null==null
while(pA != pB) {
pA = pA == nullptr ? headB : pA->next;
pB = pB == nullptr ? headA : pB->next;
}
return pA;
}