[力扣] 501. 二叉搜索树中的众数

501 二叉搜索树中的众数
给定一个有相同值的二叉搜索树(BST),找出 BST 中的所有众数(出现频率最高的元素)。
例如: 给定 BST [1,null,2,2],
返回[2].
提示:如果众数超过1个,不需考虑输出顺序
进阶:你可以不使用额外的空间吗?(假设由递归产生的隐式调用栈的开销不被计算在内)

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-mode-in-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路: 首先,二叉搜索树的中序遍历为有序序列,所以可以先遍历得到有序序列,再用哈希表统计每种数字出现的次数,最后返回出现次数最高的数字。但是这样做的空间复杂度就变成了O(n),想要达到O(1)的复杂度就不可以用哈希表。在此采用一些中间变量来记录状态,queue为二叉搜索树的中序遍历序列,num,cnt为当前的数字和出现次数,maxCount为到目前为止最大的数字出现次数,res为此时的众数,遍历完返回res。
遍历过程中要不断更新这些变量,具体实现看代码。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def findMode(self, root: TreeNode) -> List[int]:
        if root==None: return []
        def helper(root,queue):
            if root.left != None:
                helper(root.left,queue)
            queue.append(root.val)
            if root.right != None:
                helper(root.right,queue)
        queue = []
        helper(root,queue) # 中序遍历
        # print(queue)
        num, cnt = queue[0], 1  # 初始化
        maxCount, res = 1, [num]
        for i in range(1,len(queue)):
            if queue[i]==num:  # 相同数字,cnt+1, 判断是否比maxCount大,更新相应的状态
                cnt+=1
                if cnt>maxCount: # cnt>maxCount,清空res,将此时的num加入,更新maxCount
                    res = [num]
                    maxCount = cnt 
                elif cnt==maxCount: # 
                    res.append(num)
            else: # 不同数字,重新计数
                cnt, num = 1, queue[i]
                if cnt==maxCount:  # 如果maxCount为1,则一开始就将数字加入res
                    res.append(num)
        return res
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值