给定一个有相同值的二叉搜索树(BST),找出 BST 中的所有众数(出现频率最高的元素)。
假定 BST 有如下定义:
- 结点左子树中所含结点的值小于等于当前结点的值
- 结点右子树中所含结点的值大于等于当前结点的值
- 左子树和右子树都是二叉搜索树
例如:
给定 BST [1,null,2,2],
1 \ 2 / 2
返回[2].
提示:如果众数超过1个,不需考虑输出顺序
class Solution {
public:
map<int, int> check;
int MAX = 0;
vector<int> findMode(TreeNode* root)
{
Fun(root);
map<int, int>:: iterator itr;
vector<int> res;
for(itr = check.begin(); itr != check.end(); itr++)
{
if(itr ->second == MAX)
{
res.push_back(itr ->first);
}
}
return res;
}
void Fun(TreeNode* root)
{
if(root == NULL)
return ;
check[root ->val]++;
MAX = max(MAX, check[root ->val]);
Fun(root ->left);
Fun(root ->right);
}
};