链表中环的入口结点问题是一个超级经典的问题,不管是在面试中,还是考研的过程中都是一个经典问题。
通常的公认解法就是双指针(快慢指针)的解法,当然这已经的老生长谈的了。
问题
给定一个有环链表,实现一个算法返回环路的开头节点。
有环链表的定义:在链表中某个节点的next元素指向在它前面出现过的节点,则表明该链表存在环路。
示例 1:
输入:head = [3,2,0,-4], pos = 1
输出:tail connects to node index 1
解释:链表中有一个环,其尾部连接到第二个节点。
示例 2:
输入:head = [1,2], pos = 0
输出:tail connects to node index 0
解释:链表中有一个环,其尾部连接到第一个节点。
示例 3:
输入:head = [1], pos = -1
输出:no cycle
解释:链表中没有环。
算法
如果链表中有环,那么快慢指针就一定可以相遇(且一定再环上,如图上的c点),此时快指针移动过的距离是慢指针的2倍,根据图中的参数,我们可以写出以下等式:
(m+y)*2=m+xn+y //这里的xn是当相遇时快指针已经在环上循环了x次,x>=1且为整数
=> m+y=xn => m=n-y+(x-1)*n //下面解释为什么写成这种形式
接下来将快指针置于表头(此时快指针在a处,慢指针在c处),与慢指针以相同速度在链表上移动,当快指针移动到b处时,移动了m的距离,根据上面的等式可知,慢指针移动了n-y+(x-1)*n的距离。
我们来分析一下此时的慢指针在什么位置:
先移动(x-1)*n的距离,相当于在环上循环了(x-1)次,慢指针又回到了c点,然后再移动n-y的距离,如图所示,n-y正好是c点到b点的距离,说明此时慢指针也移动到了b点,即快慢指针在环路的开头节点相遇了。
js代码
/**
* Definition for singly-linked list.
* function ListNode(val) {
* this.val = val;
* this.next = null;
* }
*/
/**
* @param {ListNode} head
* @return {ListNode}
*/
var detectCycle = function(head) {
// 如果只有一个节点,或者没有环,就返回null
if (!head || !head.next) return null;
let fast = slow = head;
/**
* 因为是单链表,如果存在环的话,一定是最后一个节点的指针指向其中的某一个节点
* 所以,由于环的存在,快慢指针一定会相遇
* 类似于操场跑圈圈,跑的快的和跑的慢的,跑得快的早晚会超跑得慢的一圈
* 并在某个节点相遇
*/
while (fast && fast.next) {
fast = fast.next.next;
slow = slow.next;
if (fast == slow) {
break;
}
}
// 不是一个环,就没必要找入口了,直接返回
if (!fast || !fast.next) return null;
/**
* 一个从头节点开始,一个从相遇节点开始
* 因为:fast的移动速度是slow的两倍,则相同的时间内,fast移动的距离是slow的两倍
* 设:s为slow移动的距离,f为fast移动的距离
* 则:f=2s
* 假设:头节点到入口节点的距离为x,入口节点到相遇节点的距离为y(单向),环的周长为r,
* 设:n为f在环里面绕的圈数
* 则:s=x+y, f=x+y+n*r
* 将上式带入f=2s后,经化简可得
* x=n*r-y, 其中n*r为整数圈数的距离
* 因为:整数圈数无论跑多少圈,都会回到原点,不妨我们另n=1
* 则:x=r-y
* 所以:从头节点开始和从相遇节点开始移动,他们移动一圈的距离是相等的
* (从头节点移动的,移动距离为x;从相遇节点开始移动,移动距离为r-y)
* 所以:当他们移动相同的距离后,一定会在入口点相遇
*/
fast = head;
while (fast) {
if (fast == slow) {
break;
}
fast = fast.next;
slow = slow.next;
}
return fast;
};
另辟远道
当然还有一种方法,就是用map集合来储存链表中的数字,如果未来在移动的过程中出现了与map中相同的元素,那么这个就是环的入口。
虽然这个比较好理解,写起来也比较顺畅,但是时间复杂度和空间复杂度都不及第一种方法
var detectCycle = function(head) {
let map = new Map()
while(head){
if(map.has(head)){
return head
}else{
map.set(head,1);
head = head.next
}
}
return null
};