链表中环的入口节点

题目描述

　　一个链表中包含环，请找出该链表的环的入口结点

　　解法1：环的入口节点一定是入度为2的节点，故遍历链表，统计节点入度数，入度为2即为环入口节点

class Solution {
public:
    ListNode* EntryNodeOfLoop(ListNode* pHead)
    {
        ListNode *res=NULL;
        set<ListNode*> nodeSet;
        nodeSet.insert(pHead);
        while(pHead)
        {
            std::pair<set<ListNode*>::iterator, bool> ret=nodeSet.insert(pHead->next);
            if(!ret.second)//节点指针插入失败表示已经存在，故为入口节点
            {
                res=*(ret.first);
                break;
            }
            pHead=pHead->next;
        }
        return res;
    }
};

　　解法2：先求出环中节点数n，再两个指针一前一后，相隔为n，遍历链表，两个指针相遇点即为环的入口节点

class Solution{
public:
    int getLoopNum(ListNode* pHead)
    {
        ListNode *pre=pHead;
        ListNode *cur=pHead;
        while(cur)
        {
            cur=cur->next;
            pre=pre->next;
            if(cur)
            {
                cur=cur->next;
                if(cur==pre)
                {
                    int num=1;
                    pre=pre->next;
                    while(pre!=cur)
                    {
                        ++num;
                        pre=pre->next;
                    }
                    return num;
                }
            }
            else return 0; 
        }
        return 0;
    }
    ListNode* EntryNodeOfLoop(ListNode* pHead)
    {
        int n=getLoopNum(pHead);//得到链表中环中节点个数，如果无环，返回0
        if(n==0)return NULL;
        else
        {
            ListNode *pre=pHead;
            ListNode *cur=pHead;
            while(n--)cur=cur->next;
            while(pre!=cur)
            {
                pre=pre->next;
                cur=cur->next;
            }
            return pre;
        }
    }
};