题目描述:
给定一个链表,返回链表开始入环的第一个节点。 如果链表无环,则返回 null。
为了表示给定链表中的环,我们使用整数 pos 来表示链表尾连接到链表中的位置(索引从 0 开始)。 如果 pos 是 -1,则在该链表中没有环。注意,pos 仅仅是用于标识环的情况,并不会作为参数传递到函数中。
说明:不允许修改给定的链表。
示例:
输入:head = [3, 2, 0, -4],pos = 1
输出:返回索引为 1 的链表节点
输入:head = [1, 2],pos = 0
输出:返回索引为 0 的链表节点
输入:head = [1],pos = -1
输出:返回 null
个人解法1(HashSet):
与判断链表是否存在环类似,只需要在检测到重复节点时,直接返回该节点,即为环的第一个节点
public ListNode detectCycle(ListNode head) {
Set<ListNode> nodeSet = new HashSet<>();
ListNode cur = head;
while (cur != null) {
if (!nodeSet.contains(cur)) {
nodeSet.add(cur);
} else {
return cur;
}
cur = cur.next;
}
return null;
}
官方解法(快慢指针 + 数学推导):
以下内容转载自 LeetCode
初始时用慢指针 slow 和快指针 fast 分别指向链表头节点
慢指针每次移动一位,快指针每次移动两位。如果链表中存在环,则二者将在环中相遇。
如下图所示,设链表中环外部分的长度为 。慢指针进入环后,又走了 的距离与快指针相遇。此时,快指针已经走完了环的 圈,因此它走过的总距离:
根据题意,任意时刻,快指针走过的距离都为慢指针的 2 倍。因此有:
从相遇点到入环点的距离加上 圈的环长,恰好等于从链表头部到入环点的距离。
因此,当发现快慢指针相遇时,我们再额外使用一个指针 ptr。起始,它指向链表头部;随后,它和慢指针每次向后移动一个位置。最终,它们会在入环点相遇。
public ListNode detectCycle1(ListNode head) {
if (head == null) {
return null;
}
ListNode slow = head;
ListNode fast = head;
while (fast != null && fast.next != null) {
slow = slow.next;
fast = fast.next.next;
if (fast == slow) {
ListNode ptr = head;
while (ptr != slow) {
ptr = ptr.next;
slow = slow.next;
}
return ptr;
}
}
return null;
}