相交链表(简单)
题目描述
给你两个单链表的头节点 headA 和 headB ,请你找出并返回两个单链表相交的起始节点。如果两个链表不存在相交节点,返回 null 。
图示两个链表在节点 c1 开始相交:
解法一:
通过计算两个链表的长度差n,将长的链表从头先移动n个指针,两个指针同时向前,直到相等,此时的指针就是第一个相交的结点。
代码
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
ListNode *A=headA;
ListNode *B=headB;
int lenA=0;
int lenB=0;
while(A!=NULL){
lenA++;
A=A->next;
}
while(B!=NULL){
lenB++;
B=B->next;
}
while(lenB>lenA){
headB=headB->next;
lenB--;
}
while(lenB<lenA){
headA=headA->next;
lenA--;
}
while(headA&&headB){
if(headA==headB)
return headA;
else{
headA=headA->next;
headB=headB->next;
}
}
return NULL;
}
};
解法二(推荐):
假设A链表不包含相交部分的长度是a,B链表不包含相交部分的长度是b,相交部分的长度是c。A链表的长度就是a+c,B链表是b+c。定义两个指针p1,p2分别指向A,B的头结点,每次两个指针都同时向前移动一次,到最终相交时,两个指针走过的路程必然是相等的。 但是a,b长度不同的情况下,p1,p2只沿着原链表走一遍是不可能相交的,因此p1,p2需要循环遍历,直到两个节点相交。
循环遍历的方法是:p1先走了a+c步到达A链表尾节点,然后p1变成B链表的头节点,再走b步,共走了a+c+b步;p2先走b+c到达尾节点,然后变成A链表头节点,再走a步,共走b+c+a步;此时两个节点走过的路程相等,且在第一个相交的节点处重合。
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
ListNode* p1=headA;
ListNode* p2=headB;
while(p1!=p2){
p1=(p1==nullptr?headB:p1->next);
p2=(p2==nullptr?headA:p2->next);
}
return p1;
}
};