【剑指offer】52. 两个链表的第一个公共节点
题目:
输入两个链表,找出它们的第一个公共节点。
思路:
定义两个指针temA和temB,分别在A和B上滑动,假设两个链表的长度相交,相交部分的长度为C,则两个指针走过的距离为:L1+C+L2时,两个指着相交于交点处。
我们使用两个指针 node1
,node2
分别指向两个链表 headA
,headB
的头结点,然后同时分别逐结点遍历,当 node1
到达链表 headA
的末尾时,重新定位到链表 headB
的头结点;当 node2
到达链表 headB
的末尾时,重新定位到链表 headA
的头结点。
这样,当它们相遇时,所指向的结点就是第一个公共结点。
代码:
class Solution:
def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
node1, node2 = headA, headB
while node1 != node2:
node1 = node1.next if node1 else headB
node2 = node2.next if node2 else headA
return node1
注意: 这里假设了两条链表如果不相交时,他们相交于None。
假设链表A长度是m,链表B是n,公共部分是b,那么走过m+n+b步之后一定会相遇,返回结果。 如果没有公共部分,b=0, 那么走过m+n都指向None,返回None
可以说是非常巧妙了!
作者:z1m
链接:https://leetcode-cn.com/problems/liang-ge-lian-biao-de-di-yi-ge-gong-gong-jie-dian-lcof/solution/shuang-zhi-zhen-fa-lang-man-xiang-yu-by-ml-zimingm/
来源:力扣(LeetCode) 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。