142 环形链表II
要解决这一题的话,我们先看它的前一题,也就是判断环的存在性 (141 环形链表)
141 环形链表
题目描述
给你一个链表的头节点 head ,判断链表中是否有环。
思路
快慢指针,想象一个操场,一快一慢的两个人在操场上一直跑步(假如有环的话,没有遇到结束条件之前,都在一直循环),那么他们就一定会相遇。
结束条件设置:
- fast没有走到结尾
- 这个链表不是空链表,或者只有一个节点的链表
详细可以参考:简单易懂动画
代码
class Solution:
def hasCycle(self, head: Optional[ListNode]) -> bool:
slow = head
fast = head
while fast and fast.next: # fast 没有走到结尾,且不是单个节点
slow = slow.next
fast = fast.next.next # 要先动一下
if slow == fast:
return True # 弹出
return False
接下来重新回到142题
题目描述
给定一个链表的头节点 head ,返回链表开始入环的第一个节点。 如果链表无环,则返回 null。
思路
借鉴前一题,同样利用快慢指针对链表进行遍历。下面看一下labuladong的做法:
在快慢指针相遇时,慢指针走k步,快指针走2k步。k是快指针比慢指针多走的路,也是环的长度(第一次相遇,也就是快指针比慢指针正好多走了一圈,也就是环长度 )
设相遇点和环起点的距离是m
- 那么对于相遇点来说,慢指针走了k,所以环起点到head就是k-m。
- 巧妙的是,对于这个环来说,它的长度是k,减去m之后,就是剩下的那个弧。
为了计算,环起点到head的k-m,只需要计算剩下的弧长k-m即可。
所以,只要我们把快慢指针中的任一个重新指向 head,然后两个指针同速前进,k - m 步后一定会相遇,相遇之处就是环的起点了。
代码
第一版
class Solution:
def detectCycle(self, head: Optional[ListNode]) -> Optional[ListNode]:
slow = head
fast = head
while fast and fast.next:
slow = slow.next
fast = fast.next.next
if slow == fast:
slow = head
while fast:
slow = slow.next
fast = fast.next
if slow == fast:
return slow
return None
问题出在了第二个while,一开始就先走了一步,这显然是不对的。
第二版
由于进入到下一块的时候,就是已经确定有环了,所以不要惯性思维(while fast)
让两指针同步前进(while fast != slow),当不符合while判断的时候,退出循环,返回一个指针就行。
class Solution:
def detectCycle(self, head: Optional[ListNode]) -> Optional[ListNode]:
slow = head
fast = head
while fast and fast.next:
slow = slow.next
fast = fast.next.next
if slow == fast:
slow = head
while slow != fast:
slow = slow.next
fast = fast.next
return slow
return None
结果顺利通过
引用
主要学习的是labuladong的解法,图片里有水印,也有公众号。萌新第一次写文章,大家多见谅多提意见,谢谢朋友们!