题目
给定一个链表,返回链表开始入环的第一个节点。 如果链表无环,则返回 null。
为了表示给定链表中的环,我们使用整数 pos 来表示链表尾连接到链表中的位置(索引从 0 开始)。 如果 pos 是 -1,则在该链表中没有环。注意,pos 仅仅是用于标识环的情况,并不会作为参数传递到函数中。
说明:不允许修改给定的链表。
进阶:
- 你是否可以使用 O(1) 空间解决此题?
示例 1:
输入: head = [3,2,0,-4], pos = 1
输出: 返回索引为 1 的链表节点
解释: 链表中有一个环,其尾部连接到第二个节点。
示例 2:
输入: head = [1], pos = -1
输出: 返回 null
解释: 链表中没有环。
解决
我的思路
看到题目就知道这道题可以用快慢指针解决,但是写的过程中发现两个指针相遇的地方并不是入环点,后来想了一下感觉他们之间应该是存在着某种数学上的等式关系,但是推导了半天实在搞不出来,因为我的等式里面始终存在着一个变量,就是链表的长度,但是在这个题目中,有环,根本没办法去求链表长度,所以卡住了,后来看了题解,豁然开朗,最后一步简直神来之笔。
我的代码
public class Solution {
public ListNode detectCycle(ListNode head) {
if((head!=null && head.next==null)|| head==null) return null;
ListNode p=head,q=head;
do{
p=p.next;
if(q.next!=null) q=q.next.next;
else q=null;
if(p==q){
ListNode k=head;
while(k!=p){
k=k.next;
p=p.next;
}
return k;
}
}while(q!=null);
return null;
}
}
官方思路
我们使用两个指针,fast
与slow
。它们起始都位于链表的头部。随后,slow
指针每次向后移动一个位置,而 fast
指针向后移动两个位置。如果链表中存在环,则 fast
指针最终将再次与low
指针在环中相遇。
如下图所示,设链表中环外部分的长度为 a
。slow
指针进入环后,又走了 bb 的距离与fast
相遇。此时,fast
指针已经走完了环的 n
圈,因此它走过的总距离为
a
+
n
(
b
+
c
)
+
b
=
a
+
(
n
+
1
)
b
+
n
c
a+n(b+c)+b=a+(n+1)b+nc
a+n(b+c)+b=a+(n+1)b+nc。
根据题意,任意时刻,fast
指针走过的距离都为 slow
指针的 2 倍。因此,我们有
a + ( n + 1 ) b + n c = 2 ( a + b ) ⟹ a = c + ( n − 1 ) ( b + c ) a+(n+1)b+nc=2(a+b) \implies a=c+(n-1)(b+c) a+(n+1)b+nc=2(a+b)⟹a=c+(n−1)(b+c)
有了
a
=
c
+
(
n
−
1
)
(
b
+
c
)
a
=
c
+
(
n
−
1
)
(
b
+
c
)
a=c+(n-1)(b+c)a=c+(n−1)(b+c)
a=c+(n−1)(b+c)a=c+(n−1)(b+c) 的等量关系,我们会发现:从相遇点到入环点的距离加上 n-1
圈的环长,恰好等于从链表头部到入环点的距离。
因此,当发现slow
与fast
相遇时,我们再额外使用一个指针ptr
。起始,它指向链表头部;随后,它和slow
每次向后移动一个位置。最终,它们会在入环点相遇。
官方代码
public class Solution {
public ListNode detectCycle(ListNode head) {
if (head == null) {
return null;
}
ListNode slow = head, fast = head;
while (fast != null) {
slow = slow.next;
if (fast.next != null) {
fast = fast.next.next;
} else {
return null;
}
if (fast == slow) {
ListNode ptr = head;
while (ptr != slow) {
ptr = ptr.next;
slow = slow.next;
}
return ptr;
}
}
return null;
}
}