题目
如果一个链表中包含环,那么应该如何找出环的入口节点?从链表的头节点开始顺着next指针方向进入环的第1个节点为环的入口节点。
例如,在如图4.3所示的链表中,环的入口节点是节点3。
分析
第1步:确认是否包含环
定义两个指针并同时从链表的头节点出发,一个指针一次走一步,另一个指针一次走两步。如果链表中不包含环,走得快的指针直到抵达链表的尾节点都不会和走得慢的指针相遇。如果链表中包含环,走得快的指针在环里绕了一圈之后将会追上走得慢的指针。因此,可以根据一快一慢两个指针是否能够相遇来判断链表中是否包含环。
第2步:如何找到环的入口节点
定义两个指针来解决。先定义两个指针P1和P2,指向链表的头节点。如果链表中的环有n个节点,第1个指针P1先在链表中向前移动n步,然后两个指针以相同的速度向前移动。当第2个指针P2指向环的入口节点时,指针P1已经围绕环走了一圈又回到了入口节点。
第3步:如何得到环中节点的数目
前面在判断链表中是否有环时用到了一快一慢两个指针。如果两个指针相遇,则表明链表中存在环。两个指针之所以会相遇是因为快的指针绕环一圈追上慢的指针,因此它们相遇的节点一定是在环中。可以从这个相遇的节点出发一边继续向前移动一边计数,当再次回到这个节点时就可以得到环中节点的数目。
解
public class Test {
public static void main(String[] args) {
ListNode listNode1 = new ListNode(1);
ListNode listNode2 = new ListNode(2);
ListNode listNode3 = new ListNode(3);
ListNode listNode4 = new ListNode(4);
ListNode listNode5 = new ListNode(5);
ListNode listNode6 = new ListNode(6);
listNode1.next = listNode2;
listNode2.next = listNode3;
listNode3.next = listNode4;
listNode4.next = listNode5;
listNode5.next = listNode6;
listNode6.next = listNode3;
ListNode result = detectCycle(listNode1);
System.out.println(result.val);
}
public static ListNode detectCycle(ListNode head) {
ListNode inLoop = getNodeInLoop(head);
if (inLoop == null) {
return null;
}
int loopCount = 1;
for (ListNode n = inLoop; n.next != inLoop; n = n.next) {
loopCount++;
}
ListNode fast = head;
for (int i = 0; i < loopCount; i++) {
fast = fast.next;
}
ListNode slow = head;
while (slow != fast) {
fast = fast.next;
slow = slow.next;
}
return slow;
}
// 快慢指针找到相遇的节点
private static ListNode getNodeInLoop(ListNode head) {
if (head == null || head.next == null) {
return null;
}
ListNode slow = head.next;
ListNode fast = slow.next;
while (slow != null && fast != null) {
if (slow == fast)
return slow;
slow = slow.next;
fast = fast.next;
if (fast != null)
fast = fast.next;
}
return null;
}
}