leetcode 二叉树 501. 二叉搜索树中的众数

501. 二叉搜索树中的众数

难度简单

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

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

假定 BST 满足如下定义:

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

示例 1:

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

示例 2:

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

提示:

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

进阶:你可以不使用额外的空间吗?(假设由递归产生的隐式调用栈的开销不被计算在内)

分析:求出现频率最高的元素集合(可能存在多个总数),看到二叉搜索树马上想到中序遍历有序,使用中序遍历+双指针来解决,用 pre 指针指向前一个节点,cur 指针指向当前节点,比较两个节点的值是否相等。count 为当前统计的频率次数,maxCount为最大频率次数。一共有三种情况,当 pre 指针为空时,当前 cur 指针指向第一个元素count 为 1;当 pre 指针和 cur 指针的值相等时,count++;当 pre 指针 和 cur 指针的值不相等时,count 重新赋值为 1。若count == maxCount就将该元素加入list集合里,若count > maxCount则更新maxCount的值,清空list集合,重新添加当前元素到list集合。

class Solution {

    int count = 0;      // 统计频率
    int maxCount = 0;   // 最大频率
    TreeNode pre = null;
    List<Integer> list = new ArrayList<>();

    public int[] findMode(TreeNode root) {

        traversal(root);
        int[] res = new int[list.size()];

        for (int i = 0; i < list.size(); i++){
            res[i] = list.get(i);
        }

        return res;
    }

    // 二叉搜索树 中序遍历
    public void traversal(TreeNode cur){

        if (cur == null)
            return;

        traversal(cur.left);                // 左
                                            // 中
        if (pre == null){   // 第一个节点
            count = 1;
        }else if (pre.val == cur.val){  // 与前一个节点值相同
            count++;
        }else{  // 与前一个节点值不同
            count = 1;
        }
        pre = cur;  // 更新前一个节点

        if (count == maxCount){     // 若当前统计频率和最大频率相同,放进list中
            list.add(cur.val);
        }

        if (count > maxCount){      // 若当前统计频率大于最大频率
            maxCount = count;       // 更新最大频率
            list.clear();           // 清空list
            list.add(cur.val);
        }

        traversal(cur.right);               // 右

        return;
    }
}

执行结果:

通过

显示详情

添加备注

执行用时:0 ms, 在所有 Java 提交中击败了100.00%的用户

内存消耗:42.2 MB, 在所有 Java 提交中击败了34.90%的用户

通过测试用例:23 / 23

猜你喜欢

转载自blog.csdn.net/qq_43833393/article/details/127300250
今日推荐