题目描述
给你两个单链表的头节点 headA 和 headB ,请你找出并返回两个单链表相交的起始节点。如果两个链表不存在相交节点,返回 null 。
图示两个链表在节点 c1 开始相交:
题目数据 保证 整个链式结构中不存在环。
注意,函数返回结果后,链表必须 保持其原始结构 。
自定义评测:
评测系统 的输入如下(你设计的程序 不适用 此输入):
- intersectVal - 相交的起始节点的值。如果不存在相交节点,这一值为 0
- listA - 第一个链表
- listB - 第二个链表
- skipA - 在 listA 中(从头节点开始)跳到交叉节点的节点数
- skipB - 在 listB 中(从头节点开始)跳到交叉节点的节点数
评测系统将根据这些输入创建链式数据结构,并将两个头节点 headA 和 headB 传递给你的程序。如果程序能够正确返回相交节点,那么你的解决方案将被 视作正确答案 。
思路
也是一道往年考研题,先遍历两个链表求长度,然后用两个指针分别指向A和B的首个元素。对长的链表遍历,直到长链表待遍历节点数等于短链表长度,随后两个指针同步移动,直到二者同指。
代码
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
if(!headA || !headB)
return nullptr;
ListNode *p1 = headA;
ListNode *p2 = headB;
int m=0,n=0;
while(p1->next != nullptr){p1=p1->next; m++;}
while(p2->next != nullptr){p2=p2->next; n++;}
p1=headA;
p2=headB;
if(m>n)
while(m-->n)
p1=p1->next;
else if(m<n)
while(n-->m)
p2=p2->next;
while(p1!=nullptr&&p2!=nullptr){
if(p1==p2)
return p1;
p1=p1->next;
p2=p2->next;
}
return nullptr;
}
};