跟随carl代码随想录刷题
语言:python
面试题 02.07. 简单
链表相交
题目:给你两个单链表的头节点 headA 和 headB ,请你找出并返回
两个单链表相交的起始节点
。如果两个链表没有交点,返回 null 。
图示两个链表在节点 c1 开始相交:
题目数据 保证 整个链式结构中不存在环。
注意,函数返回结果后,链表必须 保持其原始结构 。
👉示例 1:
输入:intersectVal = 8, listA = [4,1,8,4,5], listB = [5,0,1,8,4,5], skipA = 2, skipB = 3
输出:Intersected at ‘8’
解释:相交节点的值为 8 (注意,如果两个链表相交则不能为 0)。
从各自的表头开始算起,链表 A 为 [4,1,8,4,5],链表 B 为 [5,0,1,8,4,5]。
在 A 中,相交节点前有 2 个节点;在 B 中,相交节点前有 3 个节点。
题目解析
本题思路很巧妙:
链表A
的指针curA
走完链表A
后接着走链表B
,如此往复
链表B
的指针curB
走完链表B
后接着走链表A
,如此往复
-
只要
链表A
与链表B
有交点,就一定会相遇于交点,从而跳出循环,从而返回curA所指的交点值。- eg:
- eg:
-
只要
链表A
与链表B
没有交点,就一定会同时等于None
,从而跳出循环,返回curA所指的值(此时值为None)。- eg:
- eg:
完整代码如下
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
"""
根据快慢法则,走的快的一定会追上走得慢的。
在这道题里,有的链表短,他走完了就去走另一条链表,我们可以理解为走的快的指针。
那么,只要其中一个链表走完了,就去走另一条链表的路。
如果有交点,他们最终一定会在同一个位置相遇
"""
# 如果两个链表中有一个是空链表,直接返回none
if headA is None or headB is None:
return None
# 为两个链表各自定义一个指针
curA = headA
curB = headB
# curA和curB一定会相遇。
# 只要有交点,就会相遇于交点,
# 只要没有交点,就终会同时走到None
while(curA != curB):
curA = curA.next if curA else headB # 如果A走完了,就切换到B走
curB = curB.next if curB else headA # 如果B走完了,就切换到A走
return curA