天天看点

找出有环链表中环的入口节点

今天去面试碰到这个问题,因为之前见过思路大概知道,就是忘了怎么证明,然后就在那里想啊想啊,结果还是没想出来。然而一回宿舍马上又想出来了,哎。

求法:两个指针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;
    }
};
           

继续阅读