编写一个程序,找到两个单链表相交的起始节点。
如下面的两个链表:
在节点 c1 开始相交。
示例
示例.png
输入: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) 内存。
来源:LeetCode 第160题
相关企业:
公司 | 出现时间 |
---|---|
美团 | 2020.06 |
解法:链表拼接
时间复杂度:O(n)
空间复杂度:O(1)
思路:
-
我们通常做这种题的思路是设定两个指针分别指向两个链表头部,一起向前走直到其中一个到达末端,另一个与末端距离则是两链表的 长度差。再通过长链表指针先走的方式消除长度差,最终两链表即可同时走到相交点。
-
换个方式消除长度差: 拼接两链表。
设长-短链表为 C,短-长链表为 D (分别代表长链表在前和短链表在前的拼接链表),则当 C 走到长短链表交接处时,D 走在长链表中,且与长链表头距离为 长度差;
以下图片帮助理解:当 ha == hb 时跳出,返回即可
链表拼接.png
代码:
def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
lA, lB = headA, headB
while lA != lB:
lA = lA.next if lA else headB
lB = lB.next if lB else headA
return lA
网友评论