牛客上看到这样一道题:
题目描述
一个链表中包含环,请找出该链表的环的入口结点。
思路分析:
链表中有环的题目似乎也算是一个非常经典的面试笔试题了,也是很多人喜欢问到的, 这道题其实有很多思路
思路一:
我的第一反应是用map或者set,遍历链表,把每一个节点映射到map中或者添加到set中,当要添加的节
点在map或者set中已经存在的时候,就是我们要的入口节点了。 这种方法的缺点就是牺牲了空间
思路二:
既然有环,那么我们自然会想到用两个指针,一个走的快,一个走的慢,这样肯定有重合的时候,这时候
快指针比慢指针多走了一个环 :
- 第一步,找环中相汇点。分别用p1,p2指向链表头部,p1每次走一步,p2每次走二步,直到p1==p2找到
- 在环中的相汇点。
- 第二步,找环的入口。接上步,当p1==p2时,p2所经过节点数为2x,p1所经过节点数为x,设环中有n个节点
- ,p2比p1多走一圈有2x=n+x; n=x;可以看出p1实际走了一个环的步数,再让p2指向链表头部,p1位置不变,p1,
- p2每次走一步直到p1==p2; 此时p1指向环的入口。
下面贴上两种实现:
一:map实现,set一样的思路
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};
class Solution {
public:
ListNode* EntryNodeOfLoop(ListNode* pHead)
{
if(pHead == NULL || pHead->next == NULL)
return NULL;
while(pHead != NULL){
m[pHead]++;
if(m[pHead] == 2)
return pHead;
pHead = pHead->next;
}
return NULL;
}
private:
map<ListNode*,int> m;
};
二:快慢指针实现
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};
class Solution {
public:
ListNode* EntryNodeOfLoop(ListNode* pHead)
{
if(pHead == NULL || pHead->next == NULL)
return NULL;
ListNode p1 = pHead;
ListNode p2 = pHead;
while(p2 != NULL && p2->next != NULL ){
p1 = p1->next;
p2 = p2->next->next;
if(p1 == p2){
p2 = pHead;
while(p1 != p2){
p1 = p1->next;
p2 = p2->next;
}
if(p1 == p2)
return p1;
}
}
return NULL;
}
};