LeetCode 501 二叉搜索树中的众数

题目链接

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

假定 BST 有如下定义：

结点左子树中所含结点的值小于等于当前结点的值
结点右子树中所含结点的值大于等于当前结点的值
左子树和右子树都是二叉搜索树
例如：
给定 BST [1,null,2,2],

   1
    \
     2
    /
   2

返回[2].

简单BFS，统计每个结点出现的次数即可，AC代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> findMode(TreeNode* root) {
        map<int,int>m;
        vector<int>v;
        int mx=0;
        queue<TreeNode*>q;
        q.push(root);
        while(!q.empty()){
            TreeNode* t=q.front();
            q.pop();
            if(t){
                m[t->val]++;
                if(t->left) q.push(t->left);
                if(t->right) q.push(t->right);
            }
        }
        for(auto i:m) mx=max(mx,i.second);
        for(auto i:m) if(i.second==mx) v.push_back(i.first);
        return v;
    }
};