链表中环的入口结点

问题描述

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

来源：牛客网
链接：https://www.nowcoder.com/practice/253d2c59ec3e4bc68da16833f79a38e4?tpId=13&&tqId=11208&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

C++代码实现

思路：1、set集合，循环遍历，没有则插入，有的输出，直到遍历结束
           2、使用双指针

/*
struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) :
        val(x), next(NULL) {
    }
};
*/
#include<unordered_set>
class Solution {
public:
    ListNode* EntryNodeOfLoop(ListNode* pHead)
    {
        unordered_set<ListNode*>st;
        while(pHead)
        {
            if (st.find(pHead) == st.end())
            {
                st.insert(pHead);
                pHead = pHead->next;
            }
            else
            {
                return pHead;
            }
        }
        return NULL;
    }
};

1、快慢指针，初始化fast，slow指针指向首结点
2、fast走两步，slow走一步，直到相遇
3、fast指向首结点，slow原地不动，两指针各走一步，直到相遇

1、slow首次走到B点时，第一次相遇点在C点，fast走到F点，设AB = x，BC = y
2、slow从B->C，fast从F->B->C，slow走过y距离，fast走过2y距离，所以FB+BC=2y，即FB = y
3、第一次相遇在C点，slow走过A->B->C，x+y。fast走过A->B->C->D->F->B->C，2x+2y。
4、AB+BC = x+y，AB+BC+CD+DF+FB+BC = 2x+2y，即CD+DF = x-y
5、此时fast指向首部，slow在相遇点C点，开始每个指针走一步
6、fast走到环形入口，AB = x。slow走到环形入口，CD+DF+FB = x-y+y = x

/*
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 || !fast->next)
        {
            return nullptr;
        }
        fast = pHead;
        while(fast != slow)
        {
            fast = fast->next;
            slow = slow->next;
        }
        return fast;
    }
};