链表相交

编写一个程序,找到两个单链表相交的起始节点。
如下面的两个链表:

链表相交.png

在节点 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
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。