剑指offer - 链表中环的入口

题目描述

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

要求:空间复杂度为 O(1)

思路

用两个指针,快指针一次走两步,慢指针一次走一步。如果有环,他们一定会相遇

相遇后,可以通过固定一个指针,移动另一个指针,获得环的长度 n

然后将两个指针都指向头,快指针先走 n 步,然后慢指针再走。两个指针都是一次走一步,则会在环开始的地方相遇。

# -*- coding:utf-8 -*-
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None
import collections
class Solution:
    def EntryNodeOfLoop(self, pHead):
        if not pHead:
            return None
        slow = pHead.next
        fast = pHead.next
        if fast:
            fast = fast.next
        else:
            return None
        while slow and fast:
            if slow != fast:
                slow = slow.next
                fast = fast.next
                if fast:
                    fast = fast.next
                else:
                    return None
            else:
                break
                
        if fast and slow:
           # 找到圈的长度
            n = 1
            slow = slow.next
            while slow!=fast:
                slow = slow.next
                n += 1
            
            slow = pHead
            fast = pHead
            for i in range(n-1):
                fast = fast.next
            while fast != slow:
                slow = slow.next
                fast = fast.next
            return fast
        else:
            return None
发布了82 篇原创文章 · 获赞 2 · 访问量 4345

猜你喜欢

转载自blog.csdn.net/qq_22498427/article/details/104981174