501. 二叉搜索树中的众数

Posted 2022-11-20 风车菊

501. 二叉搜索树中的众数

给你一个含重复值的二叉搜索树（BST）的根节点 root ，找出并返回 BST 中的所有 众数（即，出现频率最高的元素）。

如果树中有不止一个众数，可以按 任意顺序 返回。

假定 BST 满足如下定义：

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

来源：力扣（LeetCode）

思路

首先是一个二叉搜索树 （得出中序遍历是有序的），有重复的元素

1. 中序遍历二叉树
2. 设置一个前置节点，一直记录当前节点的前一个节点
3. 设置一个目前出现次数最多元素的次数
4. 设置一个当前遍历到的元素已经出现的次数
5. 因为中序遍历二叉搜素树有序的结果，相同的元素是排列再一起的，我们遍历当前元素时，比较一下和前一个节点是否相同，相同的话count就加1，否则立马设置count为 1
6. 计算count值结束后，和我们的maxcount做比较，相等就加入,
7. 大于，那就清空结果集，再加入当前元素（相当与，一直删除，添加）

代码

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);