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

目录结构

1.题目

2.题解

---

1.题目

给定一个有相同值的二叉搜索树（BST），找出 BST 中的所有众数（出现频率最高的元素）。

假定 BST 有如下定义：

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

示例：

给定 BST [1,null,2,2],

   1
    \
     2
    /
   2
返回[2].

提示：如果众数超过1个，不需考虑输出顺序

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-mode-in-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.题解

两次中序遍历。

• 第一次得到众数个数；
• 第二次得到具体众数。

public class Solution501 {

    @Test
    public void test501() {
        TreeNode root = new TreeNode(5);
        root.left = new TreeNode(4);
        root.left.left = new TreeNode(4);
        root.right = new TreeNode(5);
        System.out.println(Arrays.toString(findMode(root)));

    }

    private TreeNode pre = null;
    private int[] result;
    private int resultCount = 0;
    private int maxCount = 0;
    private int currCount = 0;

    public int[] findMode(TreeNode root) {
        inOrder(root);
        pre = null;
        result = new int[resultCount];
        resultCount = 0;
        currCount = 0;
        inOrder(root);
        return result;
    }

    private void inOrder(TreeNode root) {
        if (root == null) {
            return;
        }
        inOrder(root.left);
        currCount = (pre != null && pre.val == root.val) ? currCount + 1 : 1;
        if (currCount > maxCount) {
            maxCount = currCount;
            resultCount = 1;
        } else if (currCount == maxCount) {
            if (result != null) {
                result[resultCount] = root.val;
            }
            resultCount++;
        }
        pre = root;
        inOrder(root.right);
    }
}

• 时间复杂度：
• 空间复杂度：