题目描述:
编写一个程序,找到两个单链表相交的起始节点。
如下面的两个链表:
在节点 c1 开始相交。
示例:
输入:intersectVal = 8, listA = [4,1,8,4,5], listB = [5,0,1,8,4,5], skipA = 2, skipB = 3
输出:Reference of the node with value = 8
输入解释:相交节点的值为 8 (注意,如果两个链表相交则不能为 0)。从各自的表头开始算起,链表 A 为 [4,1,8,4,5],链表 B 为 [5,0,1,8,4,5]。在 A 中,相交节点前有 2 个节点;在 B 中,相交节点前有 3 个节点。
代码:
/**
* 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* a=headA;
ListNode* b=headB;
if(headA==NULL||headB==NULL)return NULL;
while(a!=b){
if(a!=NULL)a=a->next; //最后无法相遇则a=b=null时结束循环
else a=headB;
if(b!=NULL)b=b->next;
else b=headA;
}
return a;
}
};
每当a走到链表结尾null处,就跳入b链表头部,向后遍历;同样每当b走到链表结尾null处,就跳入a链表头部,向后遍历。
若a与b有交点则必相遇。因为每次a调至b或b调至a就会缩小一次a和b向后遍历的距离。(若链表a头部离交点n1个元素,链表b头部离交点n2个元素,差距为|n1-n2|,差距通过调至对方头部缩小)