0501：二叉搜索树中的众数

Ian

2016-05-15 约 393 字预计阅读 1 分钟次阅读

力扣第 501 题

题目

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

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

假定 BST 满足如下定义：

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

示例 1：

输入：root = [1,null,2,2]
输出：[2]

示例 2：

输入：root = [0]
输出：[0]

提示：

树中节点的数目在范围 [1, 10⁴] 内
-10⁵ <= Node.val <= 10⁵

进阶：你可以不使用额外的空间吗？（假设由递归产生的隐式调用栈的开销不被计算在内）

相似问题：

0098：验证二叉搜索树

分析

最简单的就是遍历得到数组，再求众数
要求不用额外空间，可以按中序遍历
- 相同元素必然相邻
- 因此维护当前元素的频次，更新结果即可

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17


class Solution:
    def findMode(self, root: Optional[TreeNode]) -> List[int]:
        sk = [root]
        res,ma = [],0
        w,pre = 0,None
        while sk:
            u = sk.pop()
            if isinstance(u,int):
                w = w+1 if u==pre else 1
                pre = u
                if w>ma:
                    res,ma = [u],w
                elif w==ma:
                    res.append(u)
            elif u:
                sk.extend([u.right,u.val,u.left])
        return res

47 ms

目录

0501：二叉搜索树中的众数

题目

分析

解答