v输入两个链表,找出它们的第一个公共节点。
方法1
若两个链表中有个为空,则返回NULL
否则采用双指针,分别指向两个链表的头节点,先同时遍历两个链表,分别记录下两个链表的长度,若某个链表更长一些,则先移动多余的长度,保证两个链表的遍历指针同时向后开始移动。碰到相同节点返回即可
/**
* 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) {
ListNode* p1 = headA;
ListNode* p2 = headB;
ListNode* p3 = NULL;
int lenA = 0;
int lenB = 0;
if(headA==NULL||headB==NULL){
return NULL;
}
while(p1!=NULL||p2!=NULL){
if(p1!=NULL){
lenA++;
p1 = p1->next;
}
if(p2!=NULL){
lenB++;
p2 = p2->next;
}
}
p1 = headA;
p2 = headB;
while(lenA!=lenB){
if(lenA>lenB){
p1 = p1->next;
lenA--;
}
if(lenB>lenA){
p2 = p2->next;
lenB--;
}
}
while(p1!=NULL&&p1!=p2){
p1 = p1->next;
p2 = p2->next;
}
return p1;
}
};
方法2
力扣官方给出的题解:
情况一:两个链表相交
链表 headA 和 headB 的长度分别是 m 和 n。假设链表 headA 的不相交部分有 a 个节点,链表 headB 的不相交部分有 b 个节点,两个链表相交的部分有 c 个节点,则有 a+c=m,b+c=n。
- 如果a=b,则两个指针会同时到达两个链表的第一个公共节点,此时返回两个链表的第一个公共节点;
如果 a≠b,则指针pA 会遍历完链表headA,指针pB 会遍历完链表headB,两个指针不会同时到达链表的尾节点,然后指针pA 移到链表headB 的头节点,指针pB 移到链表 headA 的头节点,然后两个指针继续移动,在指针 pA 移动了 a+c+b 次、指针 pB 移动了 b+c+a 次之后,两个指针会同时到达两个链表的第一个公共节点,该节点也是两个指针第一次同时指向的节点,此时返回两个链表的第一个公共节点。
情况二:两个链表不相交
链表headA 和headB 的长度分别是 m 和 n。考虑当 m=n 和 m≠n 时,两个指针分别会如何移动:
-
如果 m=n,则两个指针会同时到达两个链表的尾节点,然后同时变成空值 null,此时返回null;
-
如果 m≠n,则由于两个链表没有公共节点,两个指针也不会同时到达两个链表的尾节点,因此两个指针都会遍历完两个链表,在指针pA 移动了m+n 次、指针 pB 移动了 n+m 次之后,两个指针会同时变成空值null,此时返回 null。
/**
* 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) {
ListNode* p1 = headA;
ListNode* p2 = headB;
if(headA==NULL||headB==NULL){
return NULL;
}
while(p1!=p2){
p1 = (p1==NULL) ? headB : p1->next;
p2 = (p2==NULL) ? headA : p2->next;
}
return p1;
}
};