牛客刷题 单链表中环的入口结点

题目描述

给一个链表，若其中包含环，请找出该链表的环的入口结点，否则，输出null。

解题思路：

1 用两个快慢的指针，都从头结点开始往后走，快指针一次走两步 ，慢指针一次走一步，如果相遇，那就有环

2 如果有环，找出环中结点的个数，让快指针留在原地不动，慢指针继续往下走一步，再次相遇时，慢指针刚好绕环一周，记录这一周的步数。

3 直到环中结点的个数num后，让两个指针同时回到头结点，q指针先往前走 num个，然后两个指针再同时往前走，相遇时 就是入口结点。（就是让q先多走一圈环的结点数，那么两个再次相遇，q比p多走一圈，相遇点一定是入口点。）

 

/*
struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) :
        val(x), next(NULL) {
    }
};
*/
class Solution {
public:
    ListNode* EntryNodeOfLoop(ListNode* pHead)
    {
        if(pHead==NULL)
            return NULL;
        ListNode *p=pHead,*q=pHead->next;
        while(p!=NULL &&q!=NULL)
        {
            if(p==q)//有环
            {
                q=q->next;
                int num=1;
                while(p!=q)
                {
                    num++;
                    q=q->next;
                }
                p=q=pHead;
                for(int i=0;i<num;++i)
                {
                    q=q->next;
                }
                while(q!=p)
                {
                    p=p->next;
                    q=q->next;
                }
                return p;
            }
            else
            {
                p=p->next;
                q=q->next==NULL?NULL:q->next->next;
            }
        }
        return NULL;
    }
};