501-二叉搜索树中的众数

Posted 2020-11-25 angelica-duhurica

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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

    public int[] findMode(TreeNode root) {
        // 1.遍历一次，记录所有元素，就可以找到众数
        List<Integer> list = new ArrayList<>();
        list = scan(list, root);

        Integer[] li = list.toArray(new Integer[0]);

        HashMap<Integer, Integer> map = new HashMap<>();
        int max = 0;
        for(Integer i : li) {
            int tmp = map.getOrDefault(i, 0);
            map.put(i, tmp + 1);
            max = Math.max(max, tmp + 1);
        }
        Set<Integer> set = new HashSet<>();
        for(Integer i : map.keySet()) {
            if(map.get(i).equals(max)) {
                set.add(i);
            }
        }

        Integer[] res = set.toArray(new Integer[0]);
        int[] result = new int[res.length];
        for(int i = 0; i < res.length; i++) {
            result[i] = res[i];
        }
        return result;
    }

    private List<Integer> scan(List<Integer> list, TreeNode root) {
        if (root != null) {
            list = scan(list, root.left);
            list = scan(list, root.right);
            list.add(root.val);
        }
        return list;
    }

    int maxTimes = 0;
    int thisTimes = 0;
    List<Integer> res = new LinkedList<>();
    TreeNode pre = null;
    public int[] findMode(TreeNode root) {
        scan(root);
        int[] rr = new int[res.size()];
        for(int i = 0; i < rr.length; i++) {
            rr[i] = res.get(i);
        }
        return rr;
    }

    public void scan(TreeNode root) {
        if(root == null) {
            return;
        }

        scan(root.left);

        if(pre != null && pre.val == root.val) {
            thisTimes++;
        } else {
            thisTimes = 1;
        }
        if(thisTimes == maxTimes) {
            res.add(root.val);
        } else if(thisTimes > maxTimes) {
            maxTimes = thisTimes;
            res.clear();
            res.add(root.val);
        }
        pre = root;

        scan(root.right);
    }