题目
给定一个链表,若其中包含环,则输出环的入口节点。
若其中不包含环,则输出null。
样例
给定如上所示的链表:
[1, 2, 3, 4, 5, 6]
2
注意,这里的2表示编号是2的节点,节点编号从0开始。所以编号是2的节点就是val等于3的节点。则输出环的入口节点3.
算法
(链表,快慢指针扫描) O(n)
本题的做法比较巧妙。
用两个指针 first,second 分别从起点开始走,first 每次走一步,second每次走两步。
如果过程中 second 走到null,则说明不存在环。否则当 first 和 second 相遇后,让 first 返回起点,second 待在原地不动,然后两个指针每次分别走一步,当相遇时,相遇点就是环的入口。
可以用公式来说明:如上图所示,a是起点,b 是环的入口,c 是两个指针的第一次相遇点,ab 之间的距离是 x,bc之间的距离是 y。,我们用 z 表示从 c 点顺时针走到 b的距离。则第一次相遇时 second 所走的距离是 x+(y+z)∗n+y, n 表示圈数,同时 second 走过的距离是 first 的两倍,也就是 2(x+y),所以我们有 x+(y+z)∗n+y=2(x+y),所以 x=(n−1)×(y+z)+z。那么我们让 second 从 c 点开始走,走 x 步,会恰好走到 b 点;让 first从 a 点开始走,走 x 步,也会走到 b 点。
class Solution {
public ListNode entryNodeOfLoop(ListNode head) {
//判断当前链表是否为空或只有一个节点
if(head == null || head.next == null) return null;
ListNode first = head,second = head;
while(first != null && second != null){
first = first.next;
second = second.next;
//如果second等于null说明没有环存在
if(second != null) second = second.next;
else return null;
//找到第一次相遇的地方
if(first == second){
first = head;
while(first != null &&second != null){
first = first.next;
second = second.next;
if(first == second)
return first;
}
}
}
return null;
}
}