501. 二叉搜索树中的众数

给定一个有相同值的二叉搜索树（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

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class FindModeInBST501 {
    static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;
        TreeNode(int x) {
            this.val = x;
        }
    }
    
    private TreeNode cur;
    private int count = 1;
    private int result = 1;
    
    public int[] findMode(TreeNode root) {
        List<Integer> maxCounts = new ArrayList<>();
        inOrder(root, maxCounts);
        int[] res = new int[maxCounts.size()];
        for(int i = 0 ; i < maxCounts.size(); i++) {
            res[i] = maxCounts.get(i);
        }
        return res;      
    }
    public void inOrder(TreeNode root, List<Integer> list) {
        if(root == null) {
            return;
        }
        inOrder(root.left, list);
        if(cur != null) {
            if(root.val == cur.val) {
                count++;
                if(result == count) {
                    list.add(root.val);
                }
                if(result < count) {
                    result = count;
                    list.clear();
                    list.add(root.val);
                }
            }else {
                count = 1;
                cur = root;
                if(count == result) {
                    list.add(root.val);
                }
            }
        }else {
            cur = root;
            list.add(root.val);
        }
        inOrder(root.right, list);
    }
    
    public static void main(String[] args) {
        FindModeInBST501 t = new FindModeInBST501();
        TreeNode root = new TreeNode(1);
        root.right = new TreeNode(2);
        root.right.left = new TreeNode(2);
        System.out.println(Arrays.toString(t.findMode(root)));
    }
}