LeetCode501.二叉搜索树中的众数

题目

分析（树为普通二叉树）

如果将本题的BST树换为普通的一棵树，应该怎么做？map来保存出现的值和频率之间的关系，然后对map的value进行排序，输出最大的value所对应的key。

代码

class Solution {
public:
    vector<int>ans;  //存放最终结果
    map<int,int>m;   //key-值，value-出现的频率
    void dfs(TreeNode* root){
        if(root == NULL) return;
        dfs(root->left);
        m[root->val]++;
        dfs(root->right);
    }
    static bool cmp(pair<int,int>p1,pair<int,int>p2){
        return p1.second > p2.second;
    }
    vector<int> findMode(TreeNode* root) {
        dfs(root);
        int cnt = 0;
        vector<pair<int,int>>res(m.begin(),m.end());  //将map转换为vector对value进行排序
        sort(res.begin(),res.end(),cmp);
        for(auto it = res.begin();it != res.end();it++){
           if(it->second >= cnt){
                cnt = it->second;
                ans.push_back(it->first);
            }else break;
        }
       return ans;
    }
};

注意：

map里面存放是按照key有序存放，而不是value，如果想要对map中的value排序，不能直接用sort函数，要先将map转换为vector，再用sort进行排序。

分析（树为BST二叉树）

要利用BST树的性质，即中序遍历是递增有序的。利用BST树的常用技巧：双指针，一个指针指向上一个访问的结点，一个指针指向当前结点。普通想法应该是先遍历一遍数组，找出最大频率（maxCount），然后再重新遍历一遍数组把出现频率为maxCount的元素放进集合。（因为众数有多个）。那能否进行一次遍历就得到众数集合呢？

1. 只要当前 cnt == maxCount，就将数值加入集合

2. 当前 cnt >  maxCount，清空之前集合的内容，再加入这个元素。

代码

class Solution {
public:
    int maxCnt = 0,cnt = 0;
    TreeNode *pre = NULL;
    vector<int>res;
    void dfs(TreeNode *cur){
        if(cur == NULL) return;
        dfs(cur->left);
        
        if(pre == NULL){  //第一个结点
            cnt = 1;
        }else if(pre->val == cur->val){
            cnt++;
        }else{
            cnt = 1;
        }
        pre = cur;

        if(cnt == maxCnt){
            res.push_back(cur->val);
        }else if(cnt > maxCnt){
            maxCnt = cnt;
            res.clear();
            res.push_back(cur->val);
        }

        dfs(cur->right); 
    }
    vector<int> findMode(TreeNode* root) {
        dfs(root);
        return res;
    }
};