22王道数据结构P39T24
题目如下:
设计一个算法完成以下功能:判断一个链表是否有环,如果有,找出环的入口点并返回,否则返回NULL。
王道给出的题解
设置快慢两个指针分别为fast和slow,初始时都指向链表头 head。slow每次走一步,即slow=slow->next; fast每次走两步,即fast=fast->next->next。由于fast 比 slow走得快,如果有环,fast一定会先进入环,而slow后进入环。当两个指针都进入环后,经过若干次操作后两个指针定能在环上相遇。这样就可以判断一个链表是否有环。
如下图所示,当slow刚进入环时,fast早已进入环。因为fast每次比 slow多走一步且fast与slow的距离小于环的长度,所以fast与slow相遇时,slow所走的距离不超过环的长度。
如下图所示,设头结点到环的入口点的距离为a,环的入口点沿着环的方向到相遇点的距离为x,环长为r,相遇时fast绕过了n圈。
则有2(a+x)=a+nr+x,即 a=nr-x。显然从头结点到环的入口点的距离等于n倍的环长减去环的入口点到相遇点的距离。因此可设置两个指针,一个指向 head,一个指向相遇点,两个指针同步移动(均为一次走一步),相遇点即为环的入口点。
LNode* FindLoopStart(LNode *head){
LNode *fast=head,*slow=head;//设置快慢两个指针
while(slow!=NULL&&fast->next!=NULL){
slow=slow->next;//每次走一步
fast=fast->next->next;//每次走两步
if(slow==fast) break;//相遇
}
if(slow==NULL||fast->next==NULL)
return NULL;//没有环,返回NULL
LNode *p1=head,*p2=slow;//分别指向开始点、相遇点
while(p1!=p2){
p1=p1->next;
p2=p2->next;
}
return p1;//返回入口点
}
分析
首先这题的解法非常巧妙了的运用的快慢指针,如果有环的话进行追逐最后相遇。
但是王道这道题给出的代码是有问题的,我们编写如下代码进行测试
#include<iostream>
using namespace std;
struct LNode{
int data;
LNode *next;
};
LNode* FindLoopStart(LNode *head){
LNode *fast=head,*slow=head;//设置快慢两个指针
while(slow!=NULL&&fast->next!=NULL){
slow=slow->next;//每次走一步
fast=fast->next->next;//每次走两步
if(slow==fast) break;//相遇
}
if(slow==NULL||fast->next==NULL)
return NULL;//没有环,返回NULL
LNode *p1=head,*p2=slow;//分别指向开始点、相遇点
while(p1!=p2){
p1=p1->next;
p2=p2->next;
}
return p1;//返回入口点
}
int main(){
LNode *head=new LNode;
LNode *r=head;
int n;//链表n个元素
cin>>n;
//尾插法
for(int i=0;i<n;i++){
int dt;
cin>>dt;
LNode *node=new LNode;
node->data=dt;
r->next=node;
r=node;
}
r->next=NULL;
cout<<FindLoopStart(head)<<endl;
return 0;
}
链表中有两个结点,没有环,输出NULL
链表中有三个结点,没有环,应该输出NULL,但却死循环了
为什么呢,我们每步输出slow和fast的地址看一下
第一次slow走一步到0xa11490(第一个元素结点),fast走两步到0xa11930(第二个元素结点)
第二次slow走到0xa11930(第二个元素结点),fast应该走到第四个元素结点,但是链表总共只有三个元素,所以fast走到了0(NULL),我们在看一下循环的条件
slow!=NULL&&fast->next!=NULL
,首先第一个slow!=NUL
成立,第二个fast->next!=NULL
这里就有问题了,因为此时fast已经是空指针,所以执行fast->next
这条语句会产生空指针错误。
那么我们下面对程序进行修改
LNode* FindLoopStart(LNode *head){
LNode *fast=head,*slow=head;//设置快慢两个指针
while(slow!=NULL&&fast!=NULL&&fast->next!=NULL){
slow=slow->next;//每次走一步
fast=fast->next->next;//每次走两步
if(slow==fast) break;//相遇
}
if(slow==NULL||fast==NULL||fast->next==NULL)
return NULL;//没有环,返回NULL
LNode *p1=head,*p2=slow;//分别指向开始点、相遇点
while(p1!=p2){
p1=p1->next;
p2=p2->next;
}
return p1;//返回入口点
}
我们在条件中加入了fast!=NULL
,然后运用逻辑运算符的短路性质,巧妙避免了空指针错误。