牛客——链表中环的入口结点

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

1，设一个快指针fast每次走两个节点和一个慢指针slow每次走一个节点，如果链表中存在环，则快指针和满指针一定会相遇
2，设头节点距环入口的距离为a，环入口距快慢指针相遇节点的距离为b，快慢指针相遇节点距环入口的距离为c
3，当快慢指针第一次相遇是，慢指针走的距离为a+b，则快指针走的距离为2（a+b），快指针比满指针多走c+b距离
4，根据以上推算可得：a+b+c+b = 2(a+b)，解得：a=c
5，再设一个慢指针slow2从头节点出发，原来的slow指针从快慢指针相遇的结点出发，slow2到环入口需要走a距离，slow到环入口需要走c距离，根据4中推算，a=c，则slow2和slow一定在环入口相遇，此时的节点便是环入口

class Solution {
public:
    ListNode* EntryNodeOfLoop(ListNode* pHead)
    {
        if (pHead == NULL || pHead->next == NULL)
            return NULL;
        ListNode* fast = pHead;
        ListNode* slow = pHead;
        
        while (fast!=NULL && fast->next!=NULL)
        {
            fast = fast->next->next;
            slow = slow->next;
            if (fast == slow)
            {
                ListNode* slow2 = pHead;
                while (slow != slow2)
                {
                    slow = slow->next;
                    slow2 = slow2->next;
                }
                return slow;
            }
        }
        return NULL;
    }
};