501. 二叉搜索树中的众数
给定一个有相同值的二叉搜索树(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 {
TreeNode pre=null;
int count=1;
int max=0;
public int[] findMode(TreeNode root) {
ArrayList<Integer>list=new ArrayList<Integer>();
inOrder(root,list);
int []results=new int[list.size()];
int j=0;
for(int i:list)
results[j++]=i;
return results;
}
public void inOrder(TreeNode root,ArrayList<Integer>list)
{
if(root==null)
return;
inOrder(root.left,list);
if(pre!=null)
{
if(root.val==pre.val)
count++;
else
count=1;
pre.val=root.val;
}else
pre=new TreeNode(root.val);
if(max==count)
list.add(root.val);
if(max<count)
{
max=count;
list.clear();
list.add(root.val);
}
inOrder(root.right,list);
}
}