LeetCode:142. 环形链表 II(python)
给定一个链表,返回链表开始入环的第一个节点。 如果链表无环,则返回 null。
为了表示给定链表中的环,我们使用整数 pos 来表示链表尾连接到链表中的位置(索引从 0 开始)。 如果 pos 是 -1,则在该链表中没有环。
说明:不允许修改给定的链表。
示例 1:
输入:head = [3,2,0,-4], pos = 1
输出:tail connects to node index 1
解释:链表中有一个环,其尾部连接到第二个节点。
示例 2:
输入:head = [1,2], pos = 0
输出:tail connects to node index 0
解释:链表中有一个环,其尾部连接到第一个节点。
示例 3:
输入:head = [1], pos = -1
输出:no cycle
解释:链表中没有环。
进阶:你是否可以不用额外空间解决此题?
思路:
- 哈希,遍历链表,保存链表中每个节点,若当前节点已存在集合时,返回当前链表;若无,则返回空。
- 快慢指针(无额外空间)
- 初始时,快慢指针同时指向头结点。
- 若快指针且快指针的下一节点不为空,则慢指针走一步,快指针走两步;若某时,快慢指针相遇,标明链表有环;快慢指针相遇时,头结点与快慢指针相遇节点同步(每次走一步)向前走,直至再次相遇,该相遇节点为入环的第一个节点。
- 若快指针或快指针的下一节点为空,标明链表无环,返回空。
附代码1(Python):哈希法
class ListNode(object):
def __init__(self, x):
self.val = x
self.next = None
class Solution(object):
def detectCycle(self, head):
visited = set()
while head:
if head in visited:
return head
else:
visited.add(head)
head = head.next
return None
验证:
def buil_cycle_link(nums, pos):
'''创建循环链表'''
res = cur = ListNode(None)
meet = None
for i in range(len(nums)):
cur.next = ListNode(nums[i])
cur = cur.next
if i==pos:
meet = cur
if meet:
cur.next = meet
return res.next
nums_li = [[3,2,0,-4], [1,2], [1]]
pos_li = [1, 0, -1]
test = Solution()
for nums, pos in zip(head_li, pos_li):
head = buil_cycle_link(nums, pos) # 创建循环链表
res = test.detectCycle(head) # 寻找循环链表交点
if res:
print(res.val)
else:
print(res)
2
1
None
附代码2(Python):快慢指针法
class ListNode(object):
def __init__(self, x):
self.val = x
self.next = None
class Solution(object):
def detectCycle(self, head):
if not head:
return None
slow = fast = head
while fast and fast.next:
slow, fast = slow.next, fast.next.next
if slow == fast:
while head != slow:
head, slow = head.next, slow.next
return head
return None
验证:
def buil_cycle_link(nums, pos):
'''创建循环链表'''
res = cur = ListNode(None)
meet = None
for i in range(len(nums)):
cur.next = ListNode(nums[i])
cur = cur.next
if i==pos:
meet = cur
if meet:
cur.next = meet
return res.next
nums_li = [[3,2,0,-4], [1,2], [1]]
pos_li = [1, 0, -1]
test = Solution()
for nums, pos in zip(head_li, pos_li):
head = buil_cycle_link(nums, pos) # 创建循环链表
res = test.detectCycle(head) # 寻找循环链表交点
if res:
print(res.val)
else:
print(res)
2
1
None
注意:LeetCode提交该题时报错如下,
执行出错:Your returned value is not a ListNode type.
需要把 ListNode 类注释掉即可。