501. 二叉搜索树中的众数

题目

给你一个含重复值的二叉搜索树（BST）的根节点 root ，找出并返回 BST 中的所有 众数（即，出现频率最高的元素）。

如果树中有不止一个众数，可以按 任意顺序 返回。

假定 BST 满足如下定义：

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

思路

一棵二叉搜索树的中序遍历序列是一个非递减的有序序列，重复出现的数字一定是一个连续出现的。
用 base 记录当前的数字
用 count 记录当前数字重复的次数
用maxCount 来维护已经扫描过的数当中出现最多的那个数字的出现次数
用answer 数组记录出现的众数。每次扫描到一个新的元素：

***首先***更新 base 和 count:
如果该元素和 base 相等，那么count 自增 1；
否则将base 更新为当前数字，count 复位为 1。
***然后***更新maxCount：
如果 count=maxCount，那么说明当前的这个数字base出现的次数等于当前众数出现的次数，将 base 加入 answer 数组；
如果count>maxCount，那么说明当前的这个数字base出现的次数大于当前众数出现的次数，因此，我们需要将 maxCount 更新为 count，清空answer 数组后将base 加入answer 数组。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    List<Integer> answer = new ArrayList<>();
    int base,count,maxCount;

    public int[] findMode(TreeNode root) {
        traversal(root);
        int[] res = new int[answer.size()];
        for(int i =0; i < answer.size(); i++){
            res[i] = answer.get(i);
        }
        return res;

    }


    public void traversal(TreeNode r){
        if(r == null){
            return;
        }
        traversal(r.left);
        update(r.val);
        traversal(r.right);
    }

    public void update(int x){
        if(x == base){
            count++;
        }else{
            count = 1;
            base = x;
        }

        if(count == maxCount){
            answer.add(base);
        }
        if(count > maxCount){
            maxCount = count;
            answer.clear();
            answer.add(base);
        }
    

    }
}