题目描述
给一个链表,若其中包含环,请找出该链表的环的入口结点,否则,输出null。
方法一
/*
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};
*/
/*
1.判断环的存在:
定义两个指针,慢指针指向第一个节点,一次走一步;快指针指向慢指针后面节点,一次走两步,
如果快指针追上了慢指针,则链表包含环,快指针走到链表末尾没有碰上慢指针,则不包含环。
2.求环中节点数:
快指针与慢指针相遇一定在环中,从这个节点出发,一边走一遍计数,再回到这个节点时,可以得到环中节点数。
3.找到环的入口:
定义两个指针p1和p2,指向链表的头结点,for循环中指针p1向前走n步,n是环中的节点数,
然后进入循环两个指针一起走,当两个指针相等时停止循环
*/
class Solution {
public:
ListNode* EntryNodeOfLoop(ListNode* pHead)
{
if(pHead==nullptr)return nullptr; //空链表返回
//先计算环中结点的个数
//快慢指针相遇结点一定在环中
ListNode *pSlow=pHead,*pFast=pHead->next; //初始化两个指针
while(pFast!=NULL&&pSlow!=NULL&&pFast!=pSlow) //两个指针不为空且不相等时向后走
{
pSlow=pSlow->next;
pFast=pFast->next;
if(pFast!=nullptr) //快指针下一个节点不为空向后走
pFast=pFast->next;
} //无环退出循环时,pFast为空;有环退出循环时,pFast与pSlow相等
//开始统计环结点数
int count=1; //计数变量
ListNode *pTem=pFast->next; //临时变量,pFast从下一个开始计数
if(pFast==pSlow&&pFast!=nullptr) //退出循环时有环的情况
{
while(pTem!=pFast) //相等时,走了一圈
{
pTem=pTem->next;
++count;
}
}
else //退出循环时无环,返回
return nullptr;
//再设两指针,指向头结点
ListNode *pNode1=pHead,*pNode2=pHead;
for(int i=0;i<count;i++) //for循环,第一个指针向前走n步
{
pNode1=pNode1->next;
}
while(pNode1!=pNode2) //两个指针一起走,走到相等为止
{
pNode1=pNode1->next;
pNode2=pNode2->next;
}
return pNode1;
}
};
方法二
/*
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};
*/
/*
1.初始化vector容器,先把链表头添加到容器,
当链表不空时进入循环,指针向后走,查找指针所指元素是否在容器中,
不在容器中,把元素加入容器,继续循环;
在容器中,指针所指元素即为环的入口节点。
循环结束时,没发现重复元素,没有环。
*/
class Solution {
public:
ListNode* EntryNodeOfLoop(ListNode* pHead)
{
if(pHead == nullptr) return nullptr; //链表为空返回
vector<ListNode*> vec_list; //初始化容器
ListNode* pNode = pHead; //变量保存链表头
vec_list.push_back(pNode); //链表头加入容器
while(pNode != nullptr) //链表后面不为空,进入循环
{
pNode = pNode->next; //指针向后走
if(find(vec_list.begin(), vec_list.end(), pNode) != vec_list.end()) //判断指针所指元素是否在容器中
{
return pNode; //指针所指元素在容器中,返回指针所指元素
}
vec_list.push_back(pNode); //指针所指元素不在容器中,加入容器
}
return nullptr; //循环结束,没发现重复元素,没有环
}
};