leetcode_501. 二叉搜索树中的众数（C++）

题目

原题链接

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

假定 BST 有如下定义：

• 结点左子树中所含结点的值小于等于当前结点的值
• 结点右子树中所含结点的值大于等于当前结点的值
• 左子树和右子树都是二叉搜索树
  例如：

给定 BST [1,null,2,2],

   1
    \
     2
    /
   2
返回[2].

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

思路

其实这道题的思路很简单（但是很难想到），就是利用“二叉搜索树中序遍历序列有序”这个定理，将序列存入一个 vector 内，再求众数就很简单了，具体咋求看下面：

定义两个辅助变量：一个是遍历到当前这个数的最大连续长度num，另一个是目前找到的“众数”的出现频率maxn（自然，这是要不断更新的）
for(s[1…s.size()-1])，如果当前数与前一个数相等，num++，不相等，将 num 赋值为1
我们用num来更新maxn，如果比它大，将答案数组清空后压入当前的数，maxn = num；否则的话（也就是相等），则直接压入即可。

以上就是大体的思路，接下来是代码：

/**
 * 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:
    int num = 1, maxn = 1;
    vector<int> s, ans;
    vector<int> findMode(TreeNode* root) {
        if(!root) return ans;
        inorder(root);
        ans.push_back(s[0]);
        for(int i = 1; i < s.size(); ++ i) {
            if(s[i] == s[i-1]) num ++;
            else num = 1;
            if(num > maxn) {
                ans.clear();
                maxn = num;
                ans.push_back(s[i]);
            }else if(num == maxn) {
                ans.push_back(s[i]);
            }
        }
        return ans;
    }
    void inorder(TreeNode* root) {
        if(!root) return ;
        inorder(root -> left);
        s.push_back(root -> val);
        inorder(root -> right);
    }
};

另，这道题还可以用 Morris 算法来解，也就是用 pre 和 cur 这两个变量来代替我们的整个 s 序列，可以说是将递归换成迭代了。