欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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)
        
相关标签: LeetCode 二叉树