【算法题】返回单向链表的中间节点

最新推荐文章于 2024-09-28 23:48:31 发布

be5yond

最新推荐文章于 2024-09-28 23:48:31 发布

阅读量173

点赞数

分类专栏：算法题文章标签：链表算法数据结构

本文链接：https://blog.csdn.net/be5yond/article/details/119936589

版权

算法题专栏收录该内容

18 篇文章 0 订阅

订阅专栏

思路

如果单向链表中的节点总数为偶数，则返回第二个中间节点。

Input: 1 -> 2 -> 3 -> 4 -> 5 -> 6 -> null ，返回 4

Input: 1 -> 2 -> 3 -> 4 -> 5 -> 6 -> 7 -> null ，返回 4

同：【算法题】判断链表是否成环

使用快慢两个指针，遍历单链表。快指针步长是慢指针的2倍，快指针遍历结束时，慢指针所指的数据就是中间节点。

代码如下

In [1]: class Node:
   ...:     def __init__(self, value, next=None):
   ...:         self.value = value
   ...:         self.next = next
   ...:     def __repr__(self):
   ...:         return f'{self.value}'
   ...:

In [2]: def get_middle(node):
   ...:     slow, fast = node, node
   ...:     while fast and fast.next:
   ...:         slow = slow.next
   ...:         fast = fast.next.next
   ...:     return slow
   ...:

In [3]: node6 = Node(6)
   ...: node5 = Node(5, node6)
   ...: node4 = Node(4, node5)
   ...: node3 = Node(3, node4)
   ...: node2 = Node(2, node3)
   ...: node1 = Node(1, node2)

In [4]: get_middle(node1)
Out[4]: 4

In [5]: get_middle(node2)
Out[5]: 4

In [6]: get_middle(node4)
Out[6]: 5