题目描述
给一个链表,若其中包含环,请找出该链表的环的入口结点,否则,输出null。
思路
set容器保存每一个结点,count函数检测是否有已加入的结点,返回即可。
/*
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 NULL;
set<ListNode*> s;
ListNode* cur = pHead;//遍历结点的指针
while(cur!=NULL)
{
if(s.count(cur) == 1)
{
return cur;
}
else if(s.count(cur) == 0)
{
s.insert(cur);
cur = cur->next;
}
}
return NULL;
}
};