前置题目参考 LeeCode_141. 环形链表(暴力+龟兔赛跑)
一、介绍
1.题目描述
题目链接:https://leetcode-cn.com/problems/linked-list-cycle-ii/
给定一个链表,返回链表开始入环的第一个节点。 如果链表无环,则返回 null。
为了表示给定链表中的环,我们使用整数 pos 来表示链表尾连接到链表中的位置(索引从 0 开始)。 如果 pos 是 -1,则在该链表中没有环。注意,pos 仅仅是用于标识环的情况,并不会作为参数传递到函数中。
说明:不允许修改给定的链表。
进阶:你是否可以使用 O(1) 空间解决此题?
2.测试样例
[3,2,0,-4]
1
# tail connects to node index 1
[1,2]
0
# tail connects to node index 0
[1]
-1
# no cycle
[]
-1
# no cycle
[1,2]
-1
# no cycle
二、题解
1、暴力(set)🟢
思路:创建一个 set 保存遍历过的节点,一旦下一个节点已遍历过,说明有环。
1、定义节点 p 为空
2、当 head 不为空,判断其是否已经在set中
① 若在,说明有环,将 head 赋给 p,终止循环并输出 p
② 若不在,head后移,继续遍历
3、当 head 为空,说明无环,终止循环并输出 p
class Solution {
public:
ListNode *detectCycle(ListNode *head) {
unordered_set<ListNode*> set;
ListNode *p=nullptr;
while(head!=nullptr){
if(set.count(head)){
p=head;
break;
}
set.insert(head);
head=head->next;
}
return p;
}
};
2、龟兔赛跑(双指针)🟡
在龟兔赛跑中,兔子速度快(一次2格),乌龟速度慢(一次1格)。
- 若是链表无环,兔子必定先到达终点。【或说为:链表必存在空节点】
- 若是链表有环,无终点,兔子和乌龟必在某一处相遇。当龟==兔,说明有环
但是,若龟兔同时同点出发,将直接被判断为有环,因此让两指针一前一后。
- p 代表乌龟,从第一个结点开始,q 代表兔子,从第二个节点开始。
- 遍历过程中,一旦出现空节点,说明无环,返回false
以上是前一题的思路,对于这一题来说,新的挑战是:由于要返回从环入口开始的链表,而龟兔相遇时并不一定在入口处,因此要挖掘如何找到入口。
设起点到环入口距离为 a ,相遇点离环入口距离 b ,环长-b =c
1、相遇时,乌龟走了 x 步,x = a + b
2、设兔子在环中走完了 n 圈,兔子速度是乌龟的 2 倍。2x = a + (b + c) * n + b
3、消 x 得到 a = c + (n − 1)(b + c),即 a = 环长倍数 + c
4、不考虑完整的环,a = c。
5、因此,在相遇后,乌龟前进一格,并创建新指针 ans 位于起始处。同时移动 ans 和乌龟 p ,两者相遇点即环入口
解释:为何乌龟只走了 a+b,没有完成环?
当乌龟 p 刚到达环的入口,q 在环中某处,设它们距离为 x ,环长为 n ,兔子和乌龟速度差为1,因此兔子需要(n-x)步追上乌龟,0 ≤ n - x ≤ n,所以乌龟必在第一圈与兔子相遇。
class Solution {
public:
ListNode *detectCycle(ListNode *head) {
if(head==nullptr||head->next==nullptr) return nullptr;
// p 代表乌龟,q 代表兔子
ListNode *p=head;
ListNode *q=head->next;
// ans 用于找到环入口点
ListNode *ans=head;
while(p!=q&&q->next!=nullptr){
p=p->next;
q=q->next;
if(q->next==nullptr) return nullptr;
q=q->next;
}
// 无环
if(q->next==nullptr) return nullptr;
// 有环,继续遍历直到到达环入口
p=p->next;
while(p!=ans){
p=p->next;
ans=ans->next;
}
return ans;
}
};