LeetCode-501. 二叉搜索树中的众数

LeetCode-501. 二叉搜索树中的众数

给定一个有相同值的二叉搜索树（BST），找出 BST 中的所有众数（出现频率最高的元素）。

class Solution {
public:
    //中序遍历
    TreeNode* pre = nullptr;
    int count=1;
    int maxCount;
    vector<int> result;
    void fun(TreeNode* root){
        if(!root)return;

        fun(root->left);

        //处理逻辑
        if (pre == NULL) { // 第一个节点
            count = 1;
        } else if (pre->val == root->val) { // 与前一个节点数值相同
            count++;
        } else { // 与前一个节点数值不同
            count = 1;
        }
        pre = root; // 更新上一个节点

        if (count == maxCount) { // 如果和最大值相同，放进result中
            result.push_back(root->val);
        }

        if (count > maxCount) { // 如果计数大于最大值频率
            maxCount = count;   // 更新最大频率
            result.clear();     // 很关键的一步，不要忘记清空result，之前result里的元素都失效了
            result.push_back(root->val);
        }



        fun(root->right);
    }
    vector<int> findMode(TreeNode* root) {
        fun(root);
        return result;


    }
};

执行结果：
通过

执行用时：
12 ms, 在所有 C++ 提交中击败了92.07%的用户
内存消耗：
23.5 MB, 在所有 C++ 提交中击败了51.25%的用户
通过测试用例：
23 / 23