今天去面试碰到这个问题,因为之前见过思路大概知道,就是忘了怎么证明,然后就在那里想啊想啊,结果还是没想出来。然而一回宿舍马上又想出来了,哎。
求法:两个指针p和q,每次p走一步,q走两步,最后它们将在环上的某一个节点相遇,这时再弄一个指针r从链表表头开始,然后r走一步,p也走一步,那么最后当r==p的时候指向的节点就是环的入口节点。
证明:
假设p,q在S相遇,此时有:
p走过的节点数:L+a
q走过的节点数:L+a+n(a+b)
因为q走的节点数是p的两倍,因此得出一条等式:2L+2a=L+a+n(a+b),化简为:L=(n-1)(a+b)+b。
这条等式说明:r指针走完L个节点到达环入口,p走过的节点数为n-1个圆和b个节点,那么这时p也刚好在环入口,逆推过来就是r==p时r和p一定都同时指向环入口节点(可以证明不可能都指向其它同一个节点)。
代码:
/**
* 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 == NULL || head->next == NULL) {
return NULL;
}
ListNode* p = head;
ListNode* q = head;
p = p->next;
q = q->next->next;
while(q != NULL && q->next != NULL && p != q) {
p = p->next;
q = q->next->next;
}
//没环
if(p != q) {
return NULL;
}
ListNode* r = head;
while(r != p) {
r = r->next;
p = p->next;
}
return r;
}
};