给你一个含重复值的二叉搜索树(BST)的根节点 root ,找出并返回 BST 中的所有 众数(即,出现频率最高的元素)。
如果树中有不止一个众数,可以按 任意顺序 返回。
假定 BST 满足如下定义:
结点左子树中所含节点的值 小于等于 当前节点的值
结点右子树中所含节点的值 大于等于 当前节点的值
左子树和右子树都是二叉搜索树
来源:力扣(LeetCode)
思路
首先是一个二叉搜索树 (得出中序遍历是有序的),有重复的元素
- 中序遍历二叉树
- 设置一个前置节点,一直记录当前节点的前一个节点
- 设置一个目前出现次数最多元素的次数
- 设置一个当前遍历到的元素已经出现的次数
- 因为中序遍历二叉搜素树有序的结果,相同的元素是排列再一起的,我们遍历当前元素时,比较一下和前一个节点是否相同,相同的话count就加1,否则立马设置count为 1
- 计算count值结束后,和我们的maxcount做比较,相等就加入,
- 大于,那就清空结果集,再加入当前元素(相当与,一直删除,添加)
代码
class Solution {
// 前置节点
TreeNode pre = null;
// 当前出现最多的次数
int maxCount = 0;
// 当前数字出现的次数
int count = 0;
List<Integer> result = new ArrayList<>();
public int[] findMode(TreeNode root) {
searchBST(root);
int[] resultArr = new int[result.size()];
for(int i = 0; i < result.size() ;i++){
resultArr[i] = result.get(i);
}
return resultArr;
}
public void searchBST(TreeNode cur){
if(cur == null) return;
searchBST(cur.left);
// 当前节点为第一个节点 count 为 1
if(pre == null){
count = 1;
}else if(pre.val == cur.val){
// 当前节点和前置节点相同的话,次数加1
count++;
}else{
// 当前元素出现次数为1
count = 1;
}
pre = cur;
// 当前节点的次数和我们最大次数相等的时候 放入结果集中
if(count == maxCount){
result.add(cur.val);
}
if(count > maxCount){
// 当前元素出现的次数多与目前最大出现次数时,说明该元素晋级为目前的出现最多的数
// 我们就要清空我们结果集了,更新为当前元素
maxCount = count;
result.clear(); // 之前的都失效
result.add(cur.val);
}
searchBST(cur.right);
}
}