Leetcode 501. 二叉搜索树中的众数

给定一个有相同值的二叉搜索树(BST),找出 BST 中的所有众数(出现频率最高的元素)。

假定 BST 有如下定义:

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

例如:

给定 BST [1,null,2,2],

   1

    \

     2

    /

   2

返回[2].

提示:如果众数超过1个,不需考虑输出顺序

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

**难度**: Easy

**标签**: 树、


# -*- coding: utf-8 -*-
# @Author  : LG

"""
执行用时:56 ms, 在所有 Python3 提交中击败了99.56% 的用户
内存消耗:17.3 MB, 在所有 Python3 提交中击败了43.30% 的用户

解题思路:
    具体实现见代码注释
"""

# 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 not root:
            return []

        record = {}         # 用于保存每个节点出现的次数
        def find(root):     # 递归
            if root:
                if root.val not in record:
                    record[root.val] = 1
                else:
                    record[root.val] += 1
                find(root.left)
                find(root.right)

        find(root)
        result = {} # 各出现次数的节点
        for k, v in record.items():
            if v not in result:
                result[v] = [k]
            else:
                result[v].append(k)
        return result[sorted(result.keys())[-1]]    # 返回出现次数最多的节点