题目描述
编写一个程序,找到两个单链表相交的起始节点。
若有交点则返回c1的指针
若没有交点返回NULL
方法一
概述:双指针A和B分别从各自head开始遍历,遍历到尾部之后又从另一条链的头部开始遍历一次。附上官方题解思路。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
if (headA == nullptr || headB == nullptr) return nullptr;
ListNode* preA = headA;
ListNode* preB = headB;
while(preA != preB) { // null == null
preA = (preA == nullptr) ? headA : preA->next;
preB = (preB == nullptr) ? headB : preB->next;
}
//AB两链遍历一遍之后会自动退出循环,因为A+B长度是一样的,不用担心死循环的问题
return preA;
}
};
方法二
概述:先把长的一条的指针移动到跟短的一条的head离交点同样距离的位置处,手牵手一起走,只需遍历一次,如果有交停止遍历,返回交点,如果没有交点自然返回NULL。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
//求出每个链表的长度
int getlenth(ListNode* a){
int len=0;
ListNode *temp=a;
while(temp){
len++;
temp=temp->next;
}
return len;
}
//把长的一个链表的指针移动longer-shorter个位置
ListNode* move(int longer,int shorter,ListNode* x){
int distance=longer-shorter;
while(x&&distance--)
x=x->next;
return x;
}
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
int lena=getlenth(headA);
int lenb=getlenth(headB);
if(lena<lenb)
headB=move(lenb,lena,headB);
else
headA=move(lena,lenb,headA);
//开始遍历,如相等则return,如不相等就一直遍历下去直到结尾
while(headA&&headB){
if(headA==headB) return headA;
else{
headA=headA->next;
headB=headB->next;
}
}
return NULL;
}
};
我最开始以为只是找出第一个相等的元素,所以写了个双重遍历,以上两种方法也看了很久才想清楚,哎!