原题地址:https://leetcode-cn.com/problems/linked-list-cycle-ii/description/
题目描述:
给定一个链表,返回链表开始入环的第一个节点。 如果链表无环,则返回 null
。
说明:不允许修改给定的链表。
进阶:
你是否可以不用额外空间解决此题?
解题方案:
首先判断链表是否有环,判断方式之前考研复习时就学习到了,利用双指针,一个跑的快,一个跑的慢,他俩最后能相遇就说明链表存在环了。在记录路程时,记下的步数即为环的大小,再根据这距离往链表套环就行。
代码:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode *detectCycle(ListNode *head) {
if(!head || !head->next)
return NULL;
int count = 0;
ListNode *p, *q;
p = head; q = head->next;
while(q){
if(p == q)
break;
count ++;
p = p->next;
if(q->next)
q = q->next->next;
else
return NULL;
}
if(!q)
return NULL;
p = head; q = head;
for(int i = 0; i < count; i ++)
q = q->next;
while(q->next != p){
p = p->next;
q = q->next;
}
return p;
}
};