剑指offer-链表中环的入口结点-链表-python ***

题目描述

给一个链表，若其中包含环，请找出该链表的环的入口结点，否则，输出null。

思路

第一步，用两个快慢指针找环中相汇点。分别用slow, fast指向链表头部，slow每次走一步，fast每次走二步，直到fast == slow找到在环中的相汇点。
第二步，找环的入口。当fast == slow时，假设slow走过x个节点，则fast走过2x个节点。设环中有n个节点，因为fast比slow多走一圈（n个节点），所以有等式2x = n + x，可以推出n = x，即slow实际上走了一个环的步数。这时，我们让fast重新指向链表头部pHead，slow的位置不变，然后slow和fast一起向前每次走一步，直到fast == slow，此时两个指针相遇的节点就是环的入口。

# -*- coding:utf-8 -*-
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None
class Solution:
    def EntryNodeOfLoop(self, pHead):
        # write code here
        if  pHead is None:
            return None
        if pHead.next is None:
            return None
        p = pHead
        q = pHead.next
        while p!=q:
            if q.next is not None and q.next.next is not None:
                p = p.next
                q = q.next.next
            else:
                break
        if p ==q:
            r = pHead
            p = p.next
            while r != p:
                r = r.next
                p = p.next
            return r 
        else:
            return None