原题目
面试题 02.08. 环路检测
给定一个链表,如果它是有环链表,实现一个算法返回环路的开头节点。
有环链表的定义:在链表中某个节点的next元素指向在它前面出现过的节点,则表明该链表存在环路。
示例 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
解释:链表中没有环。
进阶:
你是否可以不用额外空间解决此题?
第一遍解法
遍历head链表,依次插入节点指针至unordered_set容器中。
插入前,判断容器中是否存在相同指针值,若存在,则认为有环路。
代码如下:
class Solution {
public:
ListNode *detectCycle(ListNode *head) {
unordered_set<ListNode*> vals;
ListNode *p = head;
while (p) {
if (vals.find(p) != vals.end()) { return p; }
vals.insert(p);
p = p->next;
}
return NULL;
}
};
时间复杂度: O(n)
空间复杂度: O(n)
不使用额外空间确实没有想出来,所以查看了网上的解法。
网上好的解法
快慢指针,
第一步,若链表中存在环路,则快慢指针必碰撞。
- 如果FastPtr刚好越过SlowPtr而不碰撞,此时假设SlowPtr处于位置 k ,FastPtr则处于位置k + 1,那么它们的前一步就都处于 k - 1。
第二步,什么时候碰撞?
第三步,上图证明了碰撞节点与环头节点的距离,与头节点与环头节点的距离相等。所以设置两个指针分别指向头结点与碰撞结点,一步步向前移动,碰撞处即为环头节点。
class Solution {
public:
ListNode *detectCycle(ListNode *head) {
ListNode *slow = head;
ListNode *fast = head;
// 快慢指针,有环时必碰撞
while (fast && fast->next) {
slow = slow->next;
fast = fast->next->next;
if (slow == fast) {
break;
}
}
// 无环路
if (fast == NULL || fast->next == NULL) {
return NULL;
}
// 关键一步,快指针与头指针同时移动,必相交于环中第一个节点
slow = head;
while (slow != fast) {
slow = slow->next;
fast = fast->next;
}
return fast;
}
};
时间复杂度: O(n)
空间复杂度: O(1)
最后的代码
与网上的代码一致。
class Solution {
public:
ListNode *detectCycle(ListNode *head) {
ListNode *slow = head;
ListNode *fast = head;
// 快慢指针,有环时必碰撞
while (fast && fast->next) {
slow = slow->next;
fast = fast->next->next;
if (slow == fast) {
break;
}
}
// 无环路
if (fast == NULL || fast->next == NULL) {
return NULL;
}
// 关键一步,快指针与头指针同时移动,必相交于环中第一个节点
slow = head;
while (slow != fast) {
slow = slow->next;
fast = fast->next;
}
return fast;
}
};
时间复杂度: O(n)
空间复杂度: O(1)
小结
- 无