876. 链表的中间结点(Python)
2019-05-30 本文已影响0人
玖月晴
题目
难度:★☆☆☆☆
类型:链表
给定一个带有头结点 head 的非空单链表,返回链表的中间结点。
如果有两个中间结点,则返回第二个中间结点。
提示
给定链表的结点数介于 1 和 100 之间。
示例
示例 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,我们返回第二个结点。
解答
方案1:统计长度
我们可以统计链表的长度,然后再次遍历,将一半长度处的结点取出。
class Solution:
def middleNode(self, head):
"""
:param head: ListNode
:return: ListNode
"""
def get_len(node):
tmp = node
count = 0
while tmp:
count += 1
tmp = tmp.next
return count
def get_node(node, idx):
tmp = node
count = 0
while count < idx:
count += 1
tmp = tmp.next
return tmp
return get_node(head, get_len(head)//2)
方案2:快慢指针
定义快慢指针,同时从头结点开始向后遍历,快指针每次走两步,慢指针每次走一步,当快指针走到末尾结点时,慢指针恰好走到中间结点。
class Solution:
def middleNode(self, head):
"""
:param head: ListNode
:return: ListNode
"""
fast = slow = head
while fast and fast.next:
fast = fast.next.next
slow = slow.next
return slow
如有疑问或建议,欢迎评论区留言~