【Day-34慢就是快】代码随想录-二叉树-搜索树中的众数

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

思路

递归法

既然是搜索树，中序遍历就是有序数组。

如何不利用数组，直接在树上进行操作呢？

弄一个指针指向前一个节点，这样每次cur（当前节点）才能和pre（前一个节点）作比较。

而且初始化的时候pre = NULL，这样当pre为NULL时候，我们就知道这是比较的第一个元素。

if (pre == NULL) { // 第一个节点
    count = 1; // 频率为1
} else if (pre->val == cur->val) { // 与前一个节点数值相同
    count++;
} else { // 与前一个节点数值不同
    count = 1;
}
pre = cur; // 更新上一个节点

但是由于树中的众数可能存在多个，我们需要用数组存放多个元素。

设置频率count和最大频率maxCount。

if (count > maxCount) { // 如果计数大于最大值
    maxCount = count;   // 更新最大频率
    result.clear();     // 很关键的一步，不要忘记清空result，之前result里的元素都失效了
    result.push_back(cur->val);
}

完整代码：

class Solution {
public:
    int maxCount = 0;
    int count = 0;
    TreeNode* pre = NULL;
    vector<int> result;
    void searchBST(TreeNode* cur){
        if(cur == NULL) return;
        searchBST(cur->left);

        if(pre == NULL)     count = 1;
        else if(pre->val == cur->val)   count++;
        else    count = 1;
        pre = cur;

        if(count == maxCount){
            result.push_back(cur->val);
        }
        if(count > maxCount){
            maxCount = count;
            result.clear();
            result.push_back(cur->val);
        }

        searchBST(cur->right);
        return;
    }

    vector<int> findMode(TreeNode* root) {
        //count = 0;
        //maxCount = 0;
        //TreeNode* pre = NULL;
        result.clear();
        searchBST(root);
        return result;
    }
};

迭代法

将中序遍历转为迭代，中间节点的处理逻辑一致。

class Solution {
public:
    vector<int> findMode(TreeNode* root) {
        stack<TreeNode*> st;
        TreeNode* cur = root;
        TreeNode* pre = NULL;
        int maxCount = 0;
        int count = 0;
        vector<int> result;
        while(cur != NULL || !st.empty()){
            if(cur != NULL){
                st.push(cur);
                cur = cur->left;
            }
            else{
                cur = st.top();
                st.pop();
                if(pre == NULL) count = 1;
                else if(pre->val == cur->val) count++;
                else count = 1;
                if(count == maxCount){
                    result.push_back(cur->val);
                }
                if(count > maxCount){
                    maxCount = count;
                    result.clear();
                    result.push_back(cur->val);
                }
                pre = cur;
                cur = cur->right;              
            }
        }
        return result;
    }
};