题目:一个链表中包含环,请找出该链表的环的入口结点。
这道题借助于前面的链表的倒数第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的判断,否则测试段错误。