给定一个有相同值的二叉搜索树(BST),找出 BST 中的所有众数(出现频率最高的元素)。
假定 BST 有如下定义:
- 结点左子树中所含结点的值小于等于当前结点的值
- 结点右子树中所含结点的值大于等于当前结点的值
- 左子树和右子树都是二叉搜索树
例如:
给定 BST [1,null,2,2]
,
1 \ 2 / 2
返回[2]
.
提示:如果众数超过1个,不需考虑输出顺序
进阶:你可以不使用额外的空间吗?(假设由递归产生的隐式调用栈的开销不被计算在内)
【分析】这里的众数就是出现最多次的数字,可以有多个,那么这道题比较直接点思路就是利用一个哈希表来记录数字和其出现次数之前的映射,然后维护一个变量max_count来记录当前最多的次数值,这样在遍历完树之后,根据这个max_count值就能把对应的元素找出来。那么用这种方法的话就不需要用到二分搜索树的性质了,随意一种遍历方式都可以,下面我们来看迭代的中序遍历的解法如下:
class Solution {
public:
vector<int> findMode(TreeNode* root) {
if(!root) return {};
TreeNode* p = root;
vector<int> res;
stack<TreeNode*> s;
int max_count = 0;
unordered_map<int,int> m;
while( p || !s.empty() ){
while(p){
s.push(p);
p = p->left;
}
p = s.top();s.pop();
max_count = max(max_count,++m[ p->val ]);
p = p->right;
}
for(auto c : m){
if(c.second == max_count){
res.push_back(c.first);
}
}
return res;
}
};
注意中序遍历的顺序是沿着最左的分支遍历到头,然后遍历最左节点的右子树,再然后才回溯到上层节点。这道题很好的练习了中序遍历的迭代实现。