题目描述
给一个链表,若其中包含环,请找出该链表的环的入口结点,否则,输出null。
题解
- 这题我们可以采用双指针解法,一快一慢指针。快指针每次跑两个element,慢指针每次跑一个。如果存在一个圈,总有一天,快指针是能追上慢指针的。
- 如下图所示,我们先找到快慢指针相遇的点,p。我们再假设,环的入口在点q,从头节点到点q距离为A,q p两点间距离为B,p q两点间距离为C。
- 因为快指针是慢指针的两倍速,且他们在p点相遇,则我们可以得到等式 2(A+B) = A+B+C+B. (此处应为:如果环前面的链表很长,而环短,那么快指针进入环以后可能转了好几圈(假设为n圈)才和慢指针相遇。但无论如何,慢指针在进入环的第一圈的时候就会和快的相遇。等式应更正为 2(A+B)= A+ nB + (n-1)C)
- 由3的等式,我们可得,C = A。
- 这时,因为我们的slow指针已经在p,我们可以让fast指针重新从头节点开始走,每次只走下一个,slow指针继续保持原来的走法,和fast同样,每次只走下一个。
- 我们期待着fast和slow指针的相遇,因为我们知道A=C,所以当他们相遇的点,一定是q了。
- 我们返回fast或者slow任意一个节点即可,因为此刻他们指向的是同一个节点,即环的起始点,q。
/*
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};
*/
class Solution {
public:
ListNode* EntryNodeOfLoop(ListNode* pHead)
{
ListNode* fast = pHead;
ListNode* slow = pHead;
while(fast && fast->next){
fast = fast->next->next;
slow = slow->next;
if(fast == slow)
break;
}
if(!fast || !fast->next)
return nullptr;
fast = pHead;
while(fast != slow){
fast = fast->next;
slow = slow->next;
}
return fast;
}
};