501. 二叉搜索树中的众数

题目描述

给你一个含重复值的二叉搜索树（BST）的根节点 root ，找出并返回 BST 中的所有 众数（即，出现频率最高的元素）。

如果树中有不止一个众数，可以按 任意顺序 返回。

假定 BST 满足如下定义：

• 结点左子树中所含节点的值 小于等于 当前节点的值
• 结点右子树中所含节点的值 大于等于 当前节点的值
• 左子树和右子树都是二叉搜索树

示例 1：

输入：root = [1,null,2,2]
输出：[2]

示例 2：

输入：root = [0]
输出：[0]

提示：

• 树中节点的数目在范围 [1, 104] 内
• -105 <= Node.val <= 105

**进阶：**你可以不使用额外的空间吗？（假设由递归产生的隐式调用栈的开销不被计算在内）

解答

/**
 * 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> res;
    // vector<int> ivec;
    int cnt = 0;
    TreeNode *pre; // 记录当前遍历节点的前一个节点，判断是否重复
    int maxCnt = 0;
    vector<int> findMode(TreeNode* root) {
        // 中序遍历过程中就可求众数（升序必然相同的数连在一起）
        cnt = 0;
        maxCnt = 0;
        res.clear();
        inorder(root);
        return res;
    }
    // 在中序遍历过程中就求众数
    void inorder(TreeNode* root)
    {
        if(root == nullptr) return ;

        inorder(root->left);
        if(pre == nullptr) cnt = 1; // 中序第一个节点
        else if(pre->val == root->val) cnt++;
        else cnt = 1; // 与前一个节点不同，重新计数
        pre = root; // 将当前节点记录到pre，对于下一次递归就是前驱节点

        if(cnt == maxCnt) res.push_back(root->val);
        if(cnt > maxCnt) // 当前节点是新的最大值，那么把原来记录的节点删除
        {
            maxCnt = cnt;
            res.clear();
            res.push_back(root->val);
        } 
        inorder(root->right);
    }
};