剑指 Offer 52. 两个链表的第一个公共节点
注意:
如果两个链表没有交点,返回 null.
在返回结果后,两个链表仍须保持原有的结构。
可假定整个链表结构中没有循环。
程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。
(来源:力扣(LeetCode))
思路:双指针
使用两个指针 node1,node2 分别指向两个链表 headA,headB 的头结点。
对于node1,它先走headA对应的分叉和重合部分(若有的话),即全部的链表A的长度,之后从headB开始,走剩下的那个分叉。
对于node2,其先走headB对应的分叉和重合部分(若有的话),即全部的链表B的长度,之后从headA开始,走剩下的那个分叉。
由于两条链表的分叉和重合部分是固定的,两指针又是同时开始,步速一致,若存在重合节点,两者最终一定可以汇合。
所以两个指针同时进行结点遍历:
(1)当 node1 到达链表 headA 的末尾时,重新定位到链表 headB 的头结点;
(2)当 node2到达链表 headB的末尾时,重新定位到链表 headA的头结点。
这样,当它们相遇时,所指向的结点就是第一个公共结点。
public class Solution {
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
if (headA == null || headB == null) {
return null;
}
ListNode node1 = headA, node2 = headB;
while (node1 != node2) {
node1 = node1 == null ? headB : node1.next;
node2 = node2 == null ? headA : node2.next;
}
return node1;
}
}