给定一个有环链表,实现一个算法返回环路的开头节点。
有环链表的定义:在链表中某个节点的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
解释:链表中没有环。
思路
判断链表中是否有环可以使用快慢指针方法来检测:
快指针每次走两步,慢指针每次走一步,如果有环的话,两个指针一定会相遇,否则快指针会走到NULL节点上;
如果有环的话将其中一个指针从头指针开始走,另一个不变,速度都改为每次走一步,当再次相遇时便就是环的入口,具体证明可以看题解
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode *detectCycle(ListNode *head) {
if(!head) head;
ListNode* pre= head, *last= head;
// 如果两个指针相遇,说明链表中有环
while(last&& last->next){
pre= pre->next;
last= last->next->next;
if(pre== last)
break;
}
// 有空说明没有环
if(!last|| !last->next) return NULL;
// 将其中一个指针从头指针开始,再次相遇时便是环的入口
last= head;
while(true){
if(last== pre)
break;
pre= pre->next;
last= last->next;
}
return last;
}
};