142. 环形链表 II
原始题目链接:https://leetcode-cn.com/problems/linked-list-cycle-ii/
给定一个链表,返回链表开始入环的第一个节点。 如果链表无环,则返回 null。
解题思路:
这道题需要数学推导等价关系式,如果有环,则推导的最终结论是:链表的头结点到入环点的距离 = 从环内相遇点到入环点的距离 + (n -1) * 环的周长。所以设置两个指针,一个快的一慢的,初始位置都在头结点,一个走两步,一个走一步,如果存在环,则第一次在环内相遇后,慢指针再需要走相遇点到入环点的距离+(n-1)倍的环周长后,就达到了入环点,此时这个距离也正好等于链表的头结点到入环点的距离,所以在第一次相遇后,为了方便找到入环点,再使用一个指针初始化为头结点,从头结点开始和慢指针一样的速度走,再次相遇就是入环点的位置。
紫色圆点为快满指针第一次相遇的点,此时快指针走了a+b+n*(b+c),慢指针走了a+b,因为快指针的速度是慢指针的2倍,所以a+b+n*(b+c) = 2(a+b),可以推导出:a=c+(n−1)(b+c),这就是上面黑色加粗的结论。有了这个结论代码就好实现了,具体看代码及注释。
代码实现:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def detectCycle(self, head: ListNode) -> ListNode:
if head is None:
return None
# 设置快慢指针
fast = head
slow = head
while True:
if (fast is None or fast.next is None):
return None
slow = slow.next
fast = fast.next.next
# 第一次相遇
if slow == fast:
break
# 准备第二次相遇
fast = head
while slow != fast:
slow = slow.next
fast = fast.next
return slow
参考文献:
https://leetcode-cn.com/problems/linked-list-cycle-ii/solution/huan-xing-lian-biao-ii-by-leetcode-solution/