[链表] LeetCode 160. 相交链表
编写一个程序,找到两个单链表相交的起始节点。
如下面的两个链表**:**
在节点 c1 开始相交。
示例 1:
输入: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 个节点。
注意:
- 如果两个链表没有交点,返回
null
. - 在返回结果后,两个链表仍须保持原有的结构。
- 可假定整个链表结构中没有循环。
- 程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。
设置两个指针pa和pb,依此后移,当pa碰到NULL是pa指向链表B的开始,当pb碰到NULL是pb指向链表A的开始,然后继续后移直到相等,相等可能是相交结点也可能是不相交返回NULL
因为两个链表长度可能不同但是A+B和B+A的长度是相同的,所以遍历A+B和B+A一定是同时结束的
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
if(!headA || !headB) return NULL;
ListNode *sA = headA;
ListNode *sB = headB;
while(headA!=headB){
if(!headA) headA = sB;
else headA = headA->next;
if(!headB) headB = sA;
else headB = headB->next;
}
return headA;
}
};