501. 二叉搜索树中的众数

最新推荐文章于 2024-10-06 20:16:20 发布

Charlesjjc

最新推荐文章于 2024-10-06 20:16:20 发布

阅读量55

点赞数

分类专栏： # 二叉树文章标签： java 算法 leetcode 数据结构深度优先

本文链接：https://blog.csdn.net/qq_37931960/article/details/119339769

版权

二叉树专栏收录该内容

28 篇文章 0 订阅

订阅专栏

501. 二叉搜索树中的众数

class Solution {
    int max = 0;
    ArrayList<Integer> ans = new ArrayList<>();
    int curmax = 1;
    TreeNode pre;
    TreeNode cur;
    public int[] findMode(TreeNode root) {
        dfs(root);
        int[] res = new int[ans.size()]; 
        for(int i = 0; i < res.length;i++){
            res[i] = ans.get(i);
        }
        return res;
    }
    public void dfs(TreeNode root){
        if(root == null) return;
        dfs(root.left);
        cur = root;
        if(pre != null){
            if(cur.val == pre.val) curmax++;
            else curmax = 1;
        }
        pre = cur;
        if(max < curmax){
            ans.clear();
            ans.add(cur.val);
            max = curmax;
        }else if(max == curmax){
            ans.add(cur.val);
        }
        dfs(root.right);
    }
}

二叉搜索树用中序遍历，利用前后指针来记录是否相等，相等计数器+1，比较历史最大众数，如果相等就在ans里加一个元素，如果出现更大的众数，则清空ans，重新放入众数，不需要额外空间，中序遍历即可完成。