501. Find Mode in Binary Search Tree [Easy] 二叉树最多节点

木木歹朱

于 2019-07-15 10:07:00 发布

阅读量138

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/qq_35319215/article/details/96921295

博客围绕二叉搜索树找众数展开，采用中序遍历得到单调递增数组，通过记录最大次数，若等于最大次数则追加，大于最大次数则清空并追加。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

501. Find Mode in Binary Search Tree

中序遍历，得到单调递增的数组，然后记录最大次数，等于最大次数：append，大于最大次数：清空+append

class Solution(object):
    def findMode(self, root):
        """
        :type root: TreeNode
        :rtype: List[int]
        """
        stack = []
        node = root
        res = []
        maxNum = curNum = 0
        pre = None
        while stack or node:
            while node:
                stack.append(node)
                node = node.left
            if stack:
                node = stack.pop()
                if node.val != pre:
                    curNum = 0
                curNum += 1
                if curNum == maxNum:
                    if not res or node.val != res[-1]:
                        res.append(node.val)
                elif curNum > maxNum:
                    maxNum = curNum
                    res = [node.val]
                pre = node.val
                node = node.right
        return res