题目:给一个链表,若其中包含环,请找出该链表的环的入口结点,否则,输出null。
解题思路:快慢指针。
- 快指针走两步,慢指针走一步,如果他们能够相遇,说明链表有环;如果不相遇,则返回None。设定首节点到环的入口点的距离为x,入口点到相遇点距离为y,相遇点到入口点为z。因为快指针是慢指针速度的二倍,因此走过的距离一定有如下关系:
- 其中左边表示的是慢指针走过距离的二倍,右边表示快指针走过的距离,n指代大于等于1的数,意味着可能环很小,走了好几圈。
- 整理上式可得:
。y+z是环的长度,z是相遇点到入口点的距离,x是首节点到入口的距离。此时相遇点的指针slow保留,重新赋值fast=head,速度为1 。当两个指针再次相等时,fast所指的节点就是入口点
代码如下:
# -*- coding:utf-8 -*-
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def EntryNodeOfLoop(self, pHead):
slow = pHead
fast = pHead
while fast and fast.next: # 能够走一步和走两步同时成立
slow = slow.next # 慢指针走一步
fast = fast.next.next # 快指针走两步
if slow == fast: # 如果两个指针能够相遇,说明一定有环
fast = pHead
while slow != fast:
slow = slow.next
fast = fast.next
return slow
return None # 如果循环完以后并没有相遇,返回None