【力扣题解】P501-二叉搜索树中的众数-Java题解

花无缺

‍博客主页:@花无缺
欢迎 点赞 收藏⭐ 留言 加关注✅!
本文由 花无缺 原创

收录于专栏 【力扣题解】


文章目录

  • 【力扣题解】P501-二叉搜索树中的众数-Java题解
    • 题目描述
    • 题解
    • 总结


【力扣题解】P501-二叉搜索树中的众数-Java题解

P501-二叉搜索树中的众数

题目描述

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

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

假定 BST 满足如下定义:

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

示例 1:

【力扣题解】P501-二叉搜索树中的众数-Java题解_第1张图片

输入:root = [1,null,2,2]
输出:[2]

示例 2:

输入:root = [0]
输出:[0]

提示:

  • 树中节点的数目在范围 [1, 104]
  • -105 <= Node.val <= 105

题解

递归

// 节点值的最大出现频率
int maxCount = Integer.MIN_VALUE;
// 统计频率
int count = 0;
List<Integer> res = new LinkedList<>();
// 保存上一个遍历的节点
TreeNode pre = null;
public int[] findMode(TreeNode root) {
    dfs(root);
    int[] a = new int[res.size()];
    for (int i = 0; i < a.length; i++) {
        a[i] = res.get(i);
    }
    return a;
}
public void dfs(TreeNode root) {
    if (root == null) {
        return;
    }
    dfs(root.left);
    // 当前节点是第一个节点, count 为 1
    if (pre == null) {
        count = 1;
    //     当前节点和前一个节点值相同, count++
    } else if (pre.val == root.val) {
        count++;
    //     当前节点和前一个节点不同, count 变为 1
    } else {
        count = 1;
    }
    // 更新 pre 节点
    pre = root;
    // 如果当前统计到的节点值次数和最大节点值次数相同
    // 就放入列表 res
    if (count == maxCount) {
        res.add(root.val);
    }
    // 如果 count > maxCount, 那么就更新 maxCount
    // 然后先清空 res, 再将当前节点值加入列表 res
    if (count > maxCount) {
        maxCount = count;
        res.clear();
        res.add(root.val);
    }
    dfs(root.right);
}

迭代

public int[] findMode(TreeNode root) {
    Deque<TreeNode> stack = new LinkedList<>();
    TreeNode cur = root;
    TreeNode pre = null;
    // 节点值的最大出现频率
    int maxCount = Integer.MIN_VALUE;
    // 统计频率
    int count = 0;
    List<Integer> res = new LinkedList<>();
    while (!stack.isEmpty() || cur != null) {
        if (cur != null) {
            stack.offerLast(cur);
            cur = cur.left;
        } else {
            cur = stack.pollLast();
            // 当前节点是第一个节点, count 为 1
            if (pre == null) {
                count = 1;
                //     当前节点和前一个节点值相同, count++
            } else if (pre.val == cur.val) {
                count++;
                //     当前节点和前一个节点不同, count 变为 1
            } else {
                count = 1;
            }
            // 更新 pre 节点
            pre = cur;
            // 如果当前统计到的节点值次数和最大节点值次数相同
            // 就放入列表 res
            if (count == maxCount) {
                res.add(cur.val);
            }
            // 如果 count > maxCount, 那么就更新 maxCount
            // 然后先清空 res, 再将当前节点值加入列表 res
            if (count > maxCount) {
                maxCount = count;
                res.clear();
                res.add(cur.val);
            }
            cur = cur.right;
        }
    }
    int[] a = new int[res.size()];
    for (int i = 0; i < a.length; i++) {
        a[i] = res.get(i);
    }
    return a;
}

时间复杂度:O(n),需要遍历二叉搜索树的所有节点,节点数为 n。

总结

这个题要求我们查找二叉搜索树中的众数,也就是出现次数最多的一个或者多个节点值,按照一般的做法,我们会将二叉搜索树的节点值放到一个数组中,对数组进行排序,然后使用双指针遍历来获取数组中的众数,但是此题我们可以直接在遍历的过程中获取众数,为什么呢?因为根据二叉搜索树的特性,我们知道二叉搜索树的中序序列是一个有序的递增序列,所以我们可以在中序遍历二叉搜索树的时候同时对节点进行操作,从而获取到众数。

同样,在处理节点时,我们采用双指针法,pre 指向上一个遍历过的节点,然后使用当前节点和 pre 指向的节点进行比较,如果相等,则统计变量 count++,否则重置为 1,当然要注意,当我们遍历第一个节点的时候,pre 为 null,这时候 count 也为 1,也就是当前节点出现了一次。

然后每一次遍历之后,我们要将当前节点频次 count 和最大频次 maxCount 作比较,只要相等,就将当前节点值加入结果列表 res,但是有可能当前节点的频次还会增多,这怎么办呢?这就要到一下步骤了,如果当前频次 count 大于 maxCount,那么就更新 maxCount,接着我们要清空 res,这样就避免了出现错误结果的情况,然后将当前节点值加入 res。

以上我也给出了迭代法的代码,和递归代码的逻辑是完全一样的。

作者:花无缺(huawuque404.com)


欢迎关注我的博客:花无缺-每一个不曾起舞的日子都是对生命的辜负~
一起进步-刷题专栏:【力扣题解】
往期精彩好文:
【全网最全爱心代码仓库】
【CSS选择器全解指南】
【HTML万字详解】
你们的点赞 收藏⭐ 留言 关注✅
是我持续创作,输出优质内容的最大动力!
谢谢!

你可能感兴趣的:(力扣题解,leetcode,java,算法,数据结构)