leetcode(91)_面试题02.08_medium_环路检测_python

环路检测

题目描述：
给定一个链表，如果它是有环链表，实现一个算法返回环路的开头节点。若环不存在，请返回 null。
如果链表中有某个节点，可以通过连续跟踪 next 指针再次到达，则链表中存在环。 为了表示给定链表中的环，我们使用整数 pos 来表示链表尾连接到链表中的位置（索引从 0 开始）。 如果 pos 是 -1，则在该链表中没有环。注意：pos 不作为参数进行传递，仅仅是为了标识链表的实际情况。
示例 ：

输入：head = [3,2,0,-4], pos = 1
输出：tail connects to node index 1
解释：链表中有一个环，其尾部连接到第二个节点。

解法

1. 较为直接的思路是用 hash 表来记录各节点及其位置，如果某节点之前已经有记录，那么它就是入口节点。
2. 如果要将空间复杂度降为常数级，那么需要设立快慢指针，如果存在环，那么它们必然会相遇。我们假设从头节点到入口节点的距离是x，从入口节点到相遇点的距离是a，从相遇点再到入口节点的距离是b，那么慢指针（一次走一步）所走的距离为x + a + m*(a+b)，快指针（一次走两步）所走的距离为x + a + n*(a+b)，从而可以得到：2*(x + a + (a+b)*m) = x + a + (a+b)*n，化简得：x = b + (a+b)(n-2m-1)。所以在快慢指针相遇后，我们只需在头节点设立一个新指针，与慢指针同时每次走一步，那么必然会在入口处相遇。

代码

# 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 not head or not head.next:
            return None
        slow, fast = head.next, head.next.next
        while slow != fast:
            if not fast or not fast.next:
                return None
            slow = slow.next
            fast = fast.next.next
        fast = head
        while fast != slow:
            fast = fast.next
            slow = slow.next
        return fast
      

测试结果

执行用时：52 ms, 在所有 Python3 提交中击败了 49.00% 的用户
内存消耗：17.9 MB, 在所有 Python3 提交中击败了 65.74% 的用户

说明

算法题来源：力扣(LeetCode)