原题链接:https://www.nowcoder.com/practice/253d2c59ec3e4bc68da16833f79a38e4
解法一:
记录每个节点的出现记录,第一个出现两次的节点为入口。
/*
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};
*/
class Solution {
public:
ListNode* EntryNodeOfLoop(ListNode* pHead) {
map<ListNode*,int> m;
while(pHead){
if(m[pHead]==1) return pHead;
m[pHead]=1;
pHead=pHead->next;
}
//这里注意:如果没有环返回nullptr
return nullptr;
}
};
解法二:
用快慢指针遍历链表,通过计算得出fast、slow相遇位置到入口距离=pHead到入口的距离,所以此时让slow和fast分别从相遇位置和起点 以相同速度出发,再次相遇时的位置即为入口。
/*
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};
*/
class Solution {
public:
ListNode* EntryNodeOfLoop(ListNode* pHead) {
ListNode *fast=pHead;
ListNode *slow=pHead;
while(fast&&fast->next){
fast=fast->next->next;
slow=slow->next;
if(fast==slow) break;
}
if(fast==nullptr||fast->next==nullptr) return nullptr;
fast=pHead;
while(fast!=slow){
fast=fast->next;
slow=slow->next;
}
return fast;
}
};