876. 链表的中间结点
题目:
给定一个头结点为 head 的非空单链表,返回链表的中间结点。
示例 1:
输入:[1,2,3,4,5]
输出:此列表中的结点 3 (序列化形式:[3,4,5])
返回的结点值为 3 。 (测评系统对该结点序列化表述是 [3,4,5])。
注意,我们返回了一个 ListNode 类型的对象 ans,这样:
ans.val = 3, ans.next.val = 4, ans.next.next.val = 5, 以及 ans.next.next.next = NULL.
示例 2:
输入:[1,2,3,4,5,6]
输出:此列表中的结点 4 (序列化形式:[4,5,6])
由于该列表有两个中间结点,值分别为 3 和 4,我们返回第二个结点。
解题:快慢指针
要返回链表的中间元素,设置快慢指针,慢指针走一步时快指针走两步。
在快指针走到链表底部的时候,慢指针正好走到链表中间
代码:
class Solution:
def middleNode(self, head: ListNode) -> ListNode:
left = right = head
while right.next:
left = left.next
right = right.next
if right.next:
right = right.next
else:
return left
return left