一、题目描述与要求
链表中环的入口结点_牛客题霸_牛客网 (nowcoder.com)
题目描述
给一个长度为n链表,若其中包含环,请找出该链表的环的入口结点,否则,返回null。
数据范围: 0≤n≤10000,1<=结点值<=10000
要求:空间复杂度O(1),时间复杂度O(n)
例如,输入{1,2},{3,4,5}时,对应的环形链表如下图所示:
可以看到环的入口结点的结点值为3,所以返回结点值为3的结点。
输入描述:
输入分为2段,第一段是入环前的链表部分,第二段是链表环的部分,后台会根据第二段是否为空将这两段组装成一个无环或者有环单链表
返回值描述:
返回链表的环的入口结点即可,我们后台程序会打印这个结点对应的结点值;若没有,则返回对应编程语言的空结点即可。
示例
示例1:
输入:{1,2},{3,4,5}
返回值:3
说明:返回环形链表入口结点,我们后台程序会打印该环形链表入口结点对应的结点值,即3
示例2:
输入:{1},{}
返回值:"null"
说明:没有环,返回对应编程语言的空结点,后台程序会打印"null"
示例3:
输入:{},{2}
返回值:2
说明:环的部分只有一个结点,所以返回该环形链表入口结点,后台程序打印该结点对应的结点值,即2
二、解题思路
如果链表有环存在的话,那么在进入到环的入口之后在继续进行遍历会再次回到环的入口,因此我们可以将链表中的结点依次存入哈希表中,一直到结点为空,如果有重复出现的结点,那么这个结点就是环的入口结点,返回当前结点,否则则没有环,返回nullptr。
三、具体代码
class Solution {
public:
ListNode* EntryNodeOfLoop(ListNode* pHead) {
unordered_set<ListNode*> set; //定义无序哈希集合
while(pHead){
if(set.count(pHead)) return pHead;//如果已经存在则直接返回
set.insert(pHead);
pHead=pHead->next;
}
return nullptr;
}
};