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