[Python手撕]二叉搜索树中的众数

给你一个含重复值的二叉搜索树(BST)的根节点 root ,找出并返回 BST 中的所有 众数(即,出现频率最高的元素)。

如果树中有不止一个众数,可以按 任意顺序 返回。

假定 BST 满足如下定义:

结点左子树中所含节点的值 小于等于 当前节点的值
结点右子树中所含节点的值 大于等于 当前节点的值
左子树和右子树都是二叉搜索树

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def findMode(self, root: Optional[TreeNode]) -> List[int]:

        def inorder(root):
            if root:
                inorder(root.left)
                
                if root.val == self.cur_val:
                    self.cur_count += 1
                else:
                    self.cur_val = root.val
                    self.cur_count = 1
                
                if self.cur_count == self.count:
                    self.res.append(self.cur_val)
                elif self.cur_count > self.count:
                    self.count = self.cur_count
                    self.res = [self.cur_val]

                inorder(root.right)

        self.cur_val = root.val
        self.cur_count = 0

        self.count = 0
        self.res = []

        inorder(root)

        return self.res

作者:Esofar

出处:https://www.cnblogs.com/DCFV/p/18469443

版权:本作品采用「署名-非商业性使用-相同方式共享 4.0 国际」许可协议进行许可。

posted @   Duancf  阅读(7)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· DeepSeek 开源周回顾「GitHub 热点速览」
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!
· AI与.NET技术实操系列(二):开始使用ML.NET
· 单线程的Redis速度为什么快?
more_horiz
keyboard_arrow_up dark_mode palette
选择主题
点击右上角即可分享
微信分享提示