算法提高之LeetCode刷题算法

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

如有疑问或建议,欢迎评论区留言~

上一篇下一篇

猜你喜欢

热点阅读