501. 二叉搜索树中的众数

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

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

假定 BST 满足如下定义:

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

示例 1:

501. 二叉搜索树中的众数_第1张图片

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

示例 2:

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

提示:

  • 树中节点的数目在范围 [1, 104] 内
  • -105 <= Node.val <= 105
class Solution {
public:
    vector ans; 
    TreeNode *pre=nullptr;
    int count=1,maxCount=0;
    void dfs(TreeNode *root)
    {
        if(root==nullptr)
            return;
        dfs(root->left);
        if(pre==nullptr)
            count=1;
        else
        {
            if(root->val==pre->val)
                count++;
            else
                count=1;
        }
        pre=root;
        if(count==maxCount)
        {
            ans.push_back(root->val);
        }
        if(count>maxCount)
        {
            maxCount=count;
            ans.clear();
            ans.push_back(root->val);
        }
        dfs(root->right);
    }
    vector findMode(TreeNode* root) {
        dfs(root);
        return ans;
    }
};

你可能感兴趣的:(leetcode题解,算法,数据结构)