给你一个含重复值的二叉搜索树(BST)的根节点
root
,找出并返回 BST 中的所有 众数(即,出现频率最高的元素)。如果树中有不止一个众数,可以按 任意顺序 返回。
假定 BST 满足如下定义:
- 结点左子树中所含节点的值 小于等于 当前节点的值
- 结点右子树中所含节点的值 大于等于 当前节点的值
- 左子树和右子树都是二叉搜索树
示例 1:
输入:root = [1,null,2,2] 输出:[2]示例 2:
输入:root = [0] 输出:[0]提示:
- 树中节点的数目在范围
[1, 104]
内-105 <= Node.val <= 105
class Solution {
public:
vector<int> ans;
TreeNode *pre=nullptr;
int count=1,maxCount=0;
void dfs(TreeNode *root)
{
if(root==nullptr)
return;
dfs(root->left);
if(pre==nullptr)
count=1;
else
{
if(root->val==pre->val)
count++;
else
count=1;
}
pre=root;
if(count==maxCount)
{
ans.push_back(root->val);
}
if(count>maxCount)
{
maxCount=count;
ans.clear();
ans.push_back(root->val);
}
dfs(root->right);
}
vector<int> findMode(TreeNode* root) {
dfs(root);
return ans;
}
};