力扣标记的难度的[简单],但看了题解并不简单,仔细一看题解是吧递归搜索变成了两层循环实现
力扣原题:501. 二叉搜索树中的众数
1. 题目描述
假定 BST 有如下定义:
结点左子树中所含结点的值小于等于当前结点的值
结点右子树中所含结点的值大于等于当前结点的值
左子树和右子树都是二叉搜索树
例如:
给定 BST [1,null,2,2],
1
2
/
2
返回[2].
提示:如果众数超过1个,不需考虑输出顺序
进阶:你可以不使用额外的空间吗?(假设由递归产生的隐式调用栈的开销不被计算在内)
2. 分析
第一想法:使用字典(键-值)数组保存所有扫描结果,之后进行一遍扫描,但这样费时费力;
第二想法:定义一个结构体,保存键-值(-- -- -_-换汤不换药),然后定义一个变量记录众数的最大个数,用以更新结构体数组;
之后的之后----结构体都不要了,直接变量表示。
3. 代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
private:
vector<int> answer;
int maxCount,base,count;
public:
void update(int cur){
if(cur == base){
count++;
}
else{
count = 1;
base = cur;
}
if(count == maxCount){
answer.push_back(cur);
}
else if(count > maxCount){
maxCount = count;
answer = vector<int>{base};//非常好用的STL
}
}
// 中序遍历
void dfs(TreeNode* root){
if(root == NULL){
return;
}
dfs(root->left);
update(root->val);
dfs(root->right);
}
vector<int> findMode(TreeNode* root) {
if(root == NULL){
return vector<int>{};
}
base = root->val;
count = 0;
dfs(root);
return answer;
}
};