501. 二叉搜索树中的众数

二叉搜索树众数

最新推荐文章于 2025-09-18 21:23:54 发布

yitahutu79

最新推荐文章于 2025-09-18 21:23:54 发布

阅读量79

点赞数

CC 4.0 BY-SA版权

分类专栏： # 力扣 # C++ 文章标签：深度优先算法

本文链接：https://blog.csdn.net/qq_40713201/article/details/127250445

C++ 同时被 2 个专栏收录

403 篇文章

订阅专栏

力扣

189 篇文章

订阅专栏

本文介绍了一种算法，用于查找含重复值的二叉搜索树中的所有众数。通过深度优先遍历的方法，实现对二叉树节点值的计数，并找出出现频率最高的元素。

给你一个含重复值的二叉搜索树（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> ans;
    int count = 0, maxnum, num;
    void cnt(int x) {
        if (x == num) {
            count++;
        } else {
            count = 1;
            num = x;
        }
        if (count > maxnum) {
            maxnum = count;
            ans.push_back(num);
            ans = vector<int> {num};
        }
        else if (count == maxnum) {
            ans.push_back(num);
        }
    }

    void dfs(TreeNode *node) {
        if (node == nullptr) return  ;
        dfs(node->left);
        cnt(node->val);
        dfs(node->right);
    }

    vector<int> findMode(TreeNode* root) {
        dfs(root);
        return ans;
    }
};