欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LeetCode 501 二叉搜索树中的众数

程序员文章站 2022-03-03 10:54:17
...

基础版,利用hash,即python中的字典,key存节点,value存出现的次数

class solution:
    def findMode(self, root):
        tmp = {}
        ans = []
        if not root:
            return ans
        def countNode(r):
            if r.val in tmp.keys():
                tmp[r.val] += 1
            else:
                tmp[r.val] = 1
            if r.left:
                countNode(r.left)
            if r.right:
                countNode(r.right)
        countNode(root)
        mode = max(tmp.values())
        for key, value in tmp.items():
            if value == mode:
                ans.append(key)
        return ans

进阶版:不使用额外的空间

tips:二叉搜索树的中序遍历是递增的,指针p,ans存储结果,遍历,出现count等于最高频率,将node存入ans,出现count大于当前最高频率,ans清空

相关标签: LeetCode