给你两个单链表的头节点 headA
和 headB
,请你找出并返回两个单链表相交的起始节点。如果两个链表不存在相交节点,返回 null
。
法1:设置两个结构体指针变量p1=headA,p2=headB,依次向后遍历,如果p1指向了A的链表末尾,那就让p1=headB,从B链表头部开始向后遍历,p2亦是如此,如果p2指向了B链表的末尾,那么就让p2重新指向A链表头部开始继续向后遍历,直到p1=p2,两个指针相逢于两个单链表相交的起始节点。
struct ListNode *getIntersectionNode(struct ListNode *headA, struct ListNode *headB) {
if(headA==NULL&&headB==NULL) return NULL;
struct ListNode *p1=headA;
struct ListNode *p2=headB;
while(p1!=p2)
{
p1=p1==NULL?headB:p1->next;
p2=p2==NULL?headA:p2->next;
}
return p1;
}
法2:先把两个链表的长度计算出来,差值用一个变量diff存储,然后设置两个结构体变量指针p1指向长的那个链表,p2指向短的那个链表,然后先遍历p1指向的链表,遍历diff个节点,这之后,p1指向的链表的接下来就和p2指向的链表的长度一样了,此时它们同时开始向后遍历,如果p1=p2,那么就找到了两个单链表相交的起始节点。
struct ListNode *getIntersectionNode(struct ListNode *headA, struct ListNode *headB)
{
if(headA==NULL&&headB==NULL) return NULL;
struct ListNode *p1=headA;
struct ListNode *p2=headB;
int len1=0,len2=0,diff=0;
while(p1!=NULL)
{
len1++;
p1=p1->next;
}
while(p2!=NULL)
{
len2++;
p2=p2->next;
}
if(len1<len2)
{
p1=headB;
p2=headA;
diff=len2-len1;
}
else
{
p1=headA;
p2=headB;
diff=len1-len2;
}
for(int i=0;i<diff;i++)
{
p1=p1->next;
}
while(p1!=NULL&&p2!=NULL)
{
if(p1==p2) return p1;
else{
p1=p1->next;
p2=p2->next;
}
}
return NULL;
}
法2可能代码看起来复杂一点,但是我觉得法2可能更容易理解一点。两种方法皆可正常运行。