LeetCode题目：二叉搜索树中的众数

二叉搜索树中的众数

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

假定 BST 有如下定义：

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

   1
    \
     2
    /
   2
   
   返回[2].

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

/**
 * 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) {
        vector<int> res;
        if(root == NULL)
            return res;
        if(root->left == NULL && root->right == NULL)
        {
            res.push_back(root->val);
            return res;
        }
        map<int, int> M;
        map<int ,int >::iterator It;; 
        TreeNode* p = root;
        stack<TreeNode*> S;
        while( p != NULL || !S.empty())
        {
            if( p != NULL)
            {
                int data = p->val;
                It=M.find(data);
                if(It==M.end())
                    M.insert(pair<int,int>(data, 0));
                else
                    It->second = M.at(data) + 1;
                S.push(p);
                p = p->left;
            }
            else
            {
                p = S.top();
                S.pop();
                p = p->right;
            }
        }
        int max_keyValue = 0;
        for(It=M.begin(); It!=M.end(); It++)
        {
            if(It->second > max_keyValue)
                max_keyValue = It->second ;
        }
        for(It=M.begin(); It!=M.end(); It++)
        {
            if(It->second == max_keyValue)   
                res.push_back(It->first);
        }
        return res;
    }
};