力扣:二叉搜索树中的众数java
流程:
中序遍历,使用搜索树的特性,中序遍历是递增数组
设置pre存储前一个节点,count为当前值的出现频次。maxcount为当前最大出现频次。
递归三部曲:
参数和返回值,返回值void,参数为节点
结束条件:根节点为null,则返回上一层
递归逻辑:
1.遍历左子树
2.中操作。判断pre指针是否为空,为空(则初始化count为1,maxcount等于count,根节点进入结果数组),不为空,(则对节点判断是否与pre的节点值相等,相等就count++,不相等则判断count重新赋值为1,在判断此时的count是否大于maxcount,大于就对count赋值为maxcount,结果集清空并输入该节点值,等于就结果集输入该值)
3.遍历右子树
代码
class Solution {
TreeNode pre;//存储前一个节点
List<Integer> result = new ArrayList<>();//结果集
int count = 0;//当前节点出现频次
int maxcount = 0;//当前最大出现频次
public int[] findMode(TreeNode root) {//主函数
TreeNode node = root;
fMode(node);//调用递归
//将List转化为int[]
int len = result.size();
int[] result1 = new int[len];
for(int i=0;i<len;i++){
result1[i] = result.get(i);
}
return result1;
}
public void fMode(TreeNode root){//递归函数
if(root == null){//结束条件,节点为空返回上一层
return;
}
fMode(root.left);//遍历左节点
if(pre != null){//pre不能为空,为空说明是第一个节点,需要对pre和count初始化
if(pre.val == root.val){//判断节点与前一节点是否值相等
count++;//相等则对当前节点值出现频次+1
}else{
count = 1;//不相等则当前节点出现频次为1
}
//当前节点出现频次是否等于最大出现频次,等于则输入结果集
if(count == maxcount) result.add(root.val);
//当前节点出现频次是否大于最大出现频次,等于则对最大频次重新赋值,清空结果集并输入本节点值
if(count > maxcount){
maxcount = count;
result.clear();
result.add(root.val);
}
}else{//需要对pre和count初始化,并输入第一个节点进入结果集
count = 1;
maxcount = 1;
result.add(root.val);
}
pre = root;//使每次遍历pre指向前一个节点
fMode(root.right);
}
}