【leetcode刷题】[简单]501. 二叉搜索树中的众数（find mode in binary search tree）-java

题目

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

假定 BST 有如下定义：

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

例如：
给定 BST [1,null,2,2],

   1
    \
     2
    /
   2

返回[2].

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

进阶：你可以不使用额外的空间吗？（假设由递归产生的隐式调用栈的开销不被计算在内）

代码模板：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int[] findMode(TreeNode root) {
        
    }
}

分析

首先这是个二叉搜索数，所以先递归root的左再看看右。用一个pre来存前一个，count为当前的数的出现次数，max为最大出现次数。如果当前等于前一个数，那么count++,否则count=1,然后去看count和max相关的。如果count>max那么，清空list，将当前数放入，并且将max赋值成count，如果count=max那么，将当前数放入list。

解答

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    int pre = 0;
    int count = 0;
    int max = 0;
    public int[] findMode(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if(root == null){
            return new int[0];
        }
        getMode(root,res);
        return res.stream().mapToInt(o->o).toArray();
    }
    public void getMode(TreeNode root,List<Integer> res){
       if(root == null){
            return;
        }
        getMode(root.left,res);
        if(pre == root.val){
            count = count+1;
        }else{
            count = 1;
        }
       if(count > max){
           res.clear();
           res.add(root.val);
           max = count;
       }else if(count == max){
           res.add(root.val);
       }
        pre = root.val;
        getMode(root.right,res);
    }
}