leetcode_501_二叉搜索树的众数

二叉搜索树的众数

描述

简单

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

假定 BST 有如下定义：

• 结点左子树中所含结点的值小于等于当前结点的值

• 结点右子树中所含结点的值大于等于当前结点的值

• 左子树和右子树都是二叉搜索树

  例如：

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

   1
    \
     2
    /
   2
返回[2].

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

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

解题

一棵二叉搜索树的中序遍历是递增的，就可以转化为在一个升序序列中查找众数

使用一个变量保存前一个数，如果当前数等于前一个数，当前的该数出现次数加1

当当前次数等于最大次数时，加入解集

当当前次数大于最大次数时，删除原有解集，加入新解

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

class Solution:
    def __init__(self):
        self.res = []
        self.pre = 0
        self.cur_count = 0
        self.max_count = 0

    def findMode(self, root: TreeNode) -> List[int]:
        if not root:
            return []
        
        self.inorder(root)
        return self.res

    def inorder(self, root):
        if not root:
            return
        self.inorder(root.left)

        self.cur_count = (self.cur_count + 1) if root.val == self.pre else 1
        if self.cur_count == self.max_count:
            self.res.append(root.val)
        elif self.cur_count > self.max_count:
            self.res.clear()
            self.res.append(root.val)
            self.max_count = self.cur_count
        
        self.pre = root.val
        self.inorder(root.right)