【leetcode】501 二叉搜索树中的众数（二叉树）

题目描述

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

假定 BST 有如下定义：

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

例如：
给定 BST [1,null,2,2],

   1
    \
     2
    /
   2

返回[2].

提示：如果众数超过1个，不需考虑输出顺序

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

思路

中序遍历是一个从小到大的排序序列，在排序序列中统计。

代码

/*
 * 二叉搜索树中的众数：中序遍历统计
 */
class Solution {
    int cnt = 1;        // 当前遍历数的计数
    int *last = nullptr;// 上一个访问节点值
    int maxCnt = 1;     // 当前众数的计数
public:
    vector<int> findMode(TreeNode* root) {
        vector<int> ret;
        if(!root) return ret;
        inOrder(root, ret);
        return ret;
    }

    void inOrder(TreeNode* root, vector<int>& ret){
        if(!root) return;
        inOrder(root->left, ret);   // 左子树

        if(last)                // 当前数字和上一个相同
            cnt = (root->val == *last)? cnt+1:1;

        if(cnt == maxCnt )      // 当前数为众数
            ret.push_back(root->val);
        else if (cnt > maxCnt){
            ret.clear();
            ret.push_back(root->val);
            maxCnt = cnt;
        }
        last = &(root->val);
        inOrder(root->right, ret);  // 右子树
    }
};