在解决链表的很多问题时,设置快慢指针是一个很好的解决思路。
这次解决两个问题:
- 链表中是否有环结构
- 求单链表的中间节点
快慢指针的另一个问题 删除链表倒数第n个结点 ,请点击查看。
链表中环的检测
class Node():
def __init__(self, data, next=None):
self.data = data
self.next = next
def has_cycle(head):
'''链表中环的检测'''
slow, fast = head, head
while fast and fast.next:
slow = slow.next
fast = fast.next.next
if slow == fast:
return True
return False
先将快慢两个指针的起始位置均设置为head节点
。
接下来,慢指针slow
每次走一步,快指针fast
每次走两步,如果链表中有环的结构,在不断的循环过程中,两个指针总会相遇。
另外,不管链表长度是奇数还是偶数,快慢两指针第一次相遇时总在head节点,大家可以画图试一下。
求链表的中间结点
def find_middle_node(head):
'''求单链表的中间结点'''
slow, fast = head, head
fast = fast.next if fast else None
while fast and fast.next:
slow = slow.next
fast = fast.next.next
return slow
将慢指针slow
起始设置为head节点
,快指针fast
起始设置为head.next
。
这里注意,如果链表为空或者链表只有一个节点(head.next == None
),则无法进入循环,直接返回原链表。
接下来,慢指针slow
每次走一步,快指针fast
每次走两步,直到循环结束,如果链表长度为奇数,此时slow
节点就是中间节点;如果链表长度为偶数,中间节点应该有两个,这里返回的slow
节点是中间两个节点的第一个,slow.next
则为中间节点的第二个。
在解决这类问题的时候,有时候只靠脑子想比较难想明白,还是要一步一步的按照代码流程画一遍图,才比较好理解。