题目
思路
这个题其实与之前环形链表的题差不多,这个麻烦的一点是需要你返回入环的第一个节点。
若我们继续用哈希表还是很简单,就是继续遍历链表,遇到的第一个重复的值即为入环的第一个节点。
这里我们看一下快慢指针的方法。
这里我们假设a为链表开头到入环第一个节点的距离。b为入环第一个节点到相遇点之间的距离,c为环剩下的距离。
我们让快慢指针都从链表开头部分开始移动,慢指针每次移一个距离,快指针每次移两个距离。
若相遇,则慢指针的移动距离为a+b,快指针移动的距离为a+n(b+c)+b。
快指针的移动距离是慢指针的2倍,由此可知a+n(b+c)+b=2(a+b)。
则a=(n-1)b+nc , a=(n-1)(b+c)+c。
所以a的距离就等于n-1圈的距离加上c。
若可以相遇,则将链表的开头定义为ptr,ptr和slow每次分别移动一格,当ptr将a走完的时候,一定会与slow在入环的第一个节点相遇,此时返回ptr即可。
代码
class Solution {
public:
ListNode *detectCycle(ListNode *head) {
ListNode* fast=head,*slow=head;
while(fast)
{
slow=slow->next;
if(fast->next==nullptr) return nullptr;
fast=fast->next->next;
//若相遇
if(fast==slow)
{
ListNode* ptr=head;
while(ptr != slow)
{
ptr=ptr->next;
slow=slow->next;
}
return ptr;
}
}
return nullptr;
}
};