给定一个有相同值的二叉搜索树(BST),找出 BST 中的所有众数(出现频率最高的元素)。
假定 BST 有如下定义:
结点左子树中所含结点的值小于等于当前结点的值
结点右子树中所含结点的值大于等于当前结点的值
左子树和右子树都是二叉搜索树
例如:
给定 BST [1,null,2,2],
1
2
/
2
返回[2].
代码片
.
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int preNum=-1;//前驱节点数值
int currentCount=0;//当前数的出现次数
int maxCount;//之前数出现的最大次数
ArrayList<Integer> mode=new ArrayList<Integer>();
public int[] findMode(TreeNode root) {
//思路:首先中序遍历整个数,因为二叉树的中序遍历是有序的
//然后当前节点与前驱节点比较,数值相同,currentCount++;数值不同,preNum=root.val;currentCount=1。结束后更新maxCount的值,即currentCount与maxCount比较,如果currentCount>maxCount,maxCount=currentCount;如果currentCount==maxCount,众数数量++;如果currentCount<maxCount,不对maxCount进行操作,继续计算下一个数值出现的次数
inOrderTree(root); //中序遍历二叉树
int[] arr=new int[mode.size()];
//得到众数的记录
for(int i=0;i<mode.size();i++){
arr[i]=mode.get(i);
}
return arr;
}
//中序遍历树的函数
public void inOrderTree(TreeNode root){
if(root!=null){
//遍历左子树
inOrderTree(root.left);
//比较当前节点值和前驱节点值,相等当前这个数的出现次数加一,不等说明出现新的数,新的数出现的系数为一
if(preNum==root.val){
currentCount++;
}else{
preNum=root.val;
currentCount=1;
}
//比较当前数的出现次数和之前的最大值。
//如果大于最大值,那么就说明这个数是众数,清空以前的记录,将此记录加入
//如果等于最大值,那么说明这个数也是众数,将这个记录加入
if(currentCount>maxCount){
maxCount=currentCount;
mode.clear();
mode.add(root.val);
}else if(currentCount==maxCount){
mode.add(root.val);
}
//遍历右子树
inOrderTree(root.right);
}
}
}