面试题56:链表中环的入口节点

      题目:一个链表中包含环,请找出该链表的环的入口结点。

    这道题借助于前面的链表的倒数第K个节点的思想,考虑快慢指针。考虑当知道了链表的环中所含的节点数目,用一个快指针先走这么多个节点,再用慢指针,两个同时走,此时他们相遇的第一个节点就是链表的入口节点。下面问题就转换为求链表环的节点数目。

    考虑链表是否成环,同样用快慢指针,当他们相遇的那个节点,肯定是环内的节点,再让它自己一步步走,等再次走到这个节点即可求到环的大小。

     代码如下:

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 pHead;
        int len=LengthOfRing(pHead);
        if(len == 0)
            return NULL;
        ListNode *slow=pHead;
        ListNode *fast=pHead;
        int num=0;
        while(num < len)
        {
            fast = fast->next;
            num++;
        }
        while(slow != fast)
        {
            slow = slow->next;
            fast = fast->next;
        }
        return slow;
    }
    int LengthOfRing(ListNode* pHead)
    {
        ListNode *fast=pHead;
        ListNode *slow=pHead;
        int num=0;
        fast = fast->next->next;
        slow = slow->next;
        if(slow == NULL)
            return 0;
        while(fast != NULL && slow != NULL)
        {
            if(fast == slow)
                break;
            slow = slow->next;
            fast = fast->next;
            if(fast != NULL)
                fast = fast->next;
        }
        if(fast == NULL)
            return 0;
        else 
        {
            ListNode *temp = slow->next;
            num++;
            while(temp != slow)
            {
                num++;
                temp = temp->next;
            }
            return num;
        }
            
    }
};
     这里要注意快慢指针那对于fast和slow是否为NULL的判断,否则测试段错误。

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页