思路
这段代码使用快慢指针法判断链表中是否有环并找到环的入口:首先让快指针每次走两步、慢指针每次走一步同时出发,如果存在环它们必定会相遇;相遇后将快指针重新指向链表头并改为每次走一步,然后快慢指针以相同速度继续前进,它们再次相遇的节点就是环的入口节点。
代码
/*
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};
*/
class Solution {
public:
ListNode* EntryNodeOfLoop(ListNode* pHead) {
if(pHead == nullptr) return nullptr;
// 定义快慢指针
ListNode* slow = pHead;
ListNode* fast = pHead;
while(fast != nullptr && fast->next != nullptr){
// 快指针是满指针的两倍速度
fast = fast->next->next;
slow = slow->next;
// 记录快慢指针第一次相遇的结点
if(slow == fast) break;
}
// 若是快指针指向null,则不存在环
if(fast == nullptr || fast->next == nullptr) return nullptr;
// 重新指向链表头部
fast = pHead;
// 与第一次相遇的结点相同速度出发,相遇结点为入口结点
while(fast != slow){
fast = fast->next;
slow = slow->next;
}
return fast;
}
};