二叉搜索树中的众数(力扣501)

根据二叉树的特性,我们使用中序遍历,确保是从小到大遍历各个节点。为了方便计数,我们使用双指针法,一旦发现当前且节点和前一个节点相同(指的是按照中序遍历的顺序的前一个),则累加该数字的次数。将出现次数最多的数字放入结果数组中。但是如何找到出现次数最多的呢?有些人可能会想遍历两次,一次找出最大次数是多少,一次找出次数等于最大次数的数字。其实有更巧妙的方法,我们不管三七二十一,一但遍历到新的数字且次数大于最大次数(注意这里的最大次数一开始就是为初始值,并不是真正的最大次数),就放入结果数组。随着遍历的进行我们该数字的次数会增加,当他大于最大次数时,就更新最大次数,同时将之前放在结果中的数字删去,放入新的数字(这个数字可能没有变,也可能变了)。不断重复这个过程,我们最终会得到真正的众数。这只是大致思路,大家可能不是很明白,可以结合我下面的代码以及注释理解。

代码及注释如下:

复制代码
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> result;
    int count = 0,maxCount = 0;
    TreeNode* pre = NULL;
    void travel(TreeNode* cur){
        //遇到空节点会直接退出递归,不会赋值给pre,因此只要pre为空,那么一定指的是叶子结点后面的空节点
        if(cur == NULL){
            return;
        }
        //递归左子树
        travel(cur -> left);
        //左右之间的全部为处理逻辑
        //pre为NULL代表我们遍历到叶子节点,这时开始给节点计数 
        if(pre == NULL || cur -> val == pre -> val){
            count++;
        }
        //一旦发现值不相等,就代表我们开始遍历新的节点数值了,count重新变为1
        if(pre!=NULL && cur -> val != pre -> val) count = 1;
        if (count == maxCount) { // 如果和最大值相同,放进result中
           result.push_back(cur->val);
        }
        if (count > maxCount) { // 如果计数大于最大值
           maxCount = count;   // 更新最大频率
           result.clear();// 很关键的一步,不要忘记清空result,之前result里的元素都失效了
           result.push_back(cur->val);
        }
        pre = cur;
        //递归右子树
        travel(cur -> right);
        return;
    }
    vector<int> findMode(TreeNode* root) {
        travel(root);
        return result;
    }
};
相关推荐
CoovallyAIHub11 小时前
SBP-YOLO:面向嵌入式悬架的轻量实时模型,实现减速带与坑洼高精度检测
深度学习·算法·计算机视觉
CoovallyAIHub12 小时前
医药、零件、饮料瓶盖……SuperSimpleNet让质检“即插即用”
深度学习·算法·计算机视觉
dragoooon3412 小时前
[优选算法专题二滑动窗口——串联所有单词的子串]
数据结构·c++·学习·算法·leetcode·学习方法
刃神太酷啦12 小时前
C++ 异常处理机制:从基础到实践的全面解析----《Hello C++ Wrold!》(20)--(C/C++)
java·c语言·开发语言·c++·qt·算法·leetcode
Brookty13 小时前
【算法】双指针(二)复写零
学习·算法
胖达不服输13 小时前
「日拱一码」081 机器学习——梯度增强特征选择GBFS
人工智能·python·算法·机器学习·梯度增强特征选择·gbfs
初级炼丹师(爱说实话版)13 小时前
2025算法八股——深度学习——优化器小结
人工智能·深度学习·算法
努力的小帅13 小时前
C++_哈希
开发语言·c++·学习·算法·哈希算法·散列表
Christo314 小时前
TFS-2023《Fuzzy Clustering With Knowledge Extraction and Granulation》
人工智能·算法·机器学习·支持向量机
过河卒_zh156676614 小时前
AI内容标识新规实施后,大厂AI用户协议有何变化?(二)百度系
人工智能·算法·aigc·算法备案·生成合成类算法备案