题目
给定一个链表,返回链表开始入环的第一个节点。 如果链表无环,则返回 null。
为了表示给定链表中的环,我们使用整数 pos 来表示链表尾连接到链表中的位置(索引从 0 开始)。 如果 pos 是 -1,则在该链表中没有环。
说明:不允许修改给定的链表。
示例 1:
输入:head = [3,2,0,-4], pos = 1
输出:tail connects to node index 1
解释:链表中有一个环,其尾部连接到第二个节点。
示例 2:
输入:head = [1,2], pos = 0
输出:tail connects to node index 0
解释:链表中有一个环,其尾部连接到第一个节点。
示例 3:
输入:head = [1], pos = -1
输出:no cycle
解释:链表中没有环。
解题思路
这道题是环形链表(判断链表中是否有环)和链表中环的入口的结合。如果对空间没有限制,直接用dict检查即可。如果有限制,则用快慢指针。
首先判断链表是否有环:
快慢2个指针,慢指针1次走1步,快指针1次走2步,如果快慢指针相遇,则说明链表有环。
证明:
如果链表有环,在环中,由于快指针比慢指针每次多走1步,所以以慢指针作为参考系,如果有环,则在环中慢指针不动,快指针每次走1步。因此必然能在环中的节点相遇,并且快指针多走的路恰好是1个环的长度。
其次找出环的长度:
上述快慢指针相遇时,走过的节点数量差就是环的长度
证明:
记慢指针走过的节点数为
s
s
s,快指针走过的节点数为
f
f
f,由于快指针速度是慢指针2倍,所以有
f
=
2
s
f=2s
f=2s。又因为快指针比慢指针多走了1
个环的长度l
,所以有
f
=
s
+
l
f=s+l
f=s+l,即
f
=
2
l
,
s
=
l
f=2l, s=l
f=2l,s=l。第一次相遇时两者走过的节点数之差就是环的长度
最后找环的入口:
如果从链表的头指针开始走,设不在环中的节点为
x
x
x个,那么从链表头部开始,每走
x
+
n
l
x+nl
x+nl个节点,都会走到环的入口。上面的快慢指针相遇的地方,已经走了
n
l
nl
nl个节点,因此再走
x
x
x个节点,就会找到环的入口。怎么再走
x
x
x个节点呢?2个指针,一个从链表头部开始走,一个从快慢指针相遇的地方开始走,同时走
x
x
x个节点,此时两个指针再次相遇,就找到了环的入口
代码
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def detectCycle(self, head: Optional[ListNode]) -> Optional[ListNode]:
slow, fast = head, head
loop_exist = False
while fast and fast.next:
slow = slow.next
fast = fast.next.next
if fast == slow:
loop_exist = True
break
if not loop_exist:
return None
slow = head
while slow != fast:
slow = slow.next
fast =fast.next
return slow