leetcode_501_二叉搜索树的众数
程序员文章站
2022-05-20 16:11:28
...
二叉搜索树的众数
描述
简单
给定一个有相同值的二叉搜索树(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)